Update chat_model.py
This commit is contained in:
parent
a08ba254c8
commit
896c656185
|
@ -29,7 +29,6 @@ class ChatModel:
|
|||
else:
|
||||
raise NotImplementedError("Unknown backend: {}".format(model_args.infer_backend))
|
||||
|
||||
self.system_message = generating_args.system_message or None
|
||||
self._loop = asyncio.new_event_loop()
|
||||
self._thread = Thread(target=_start_background_loop, args=(self._loop,), daemon=True)
|
||||
self._thread.start()
|
||||
|
@ -64,7 +63,6 @@ class ChatModel:
|
|||
image: Optional["NDArray"] = None,
|
||||
**input_kwargs,
|
||||
) -> Generator[str, None, None]:
|
||||
system = system or self.system_message
|
||||
generator = self.astream_chat(messages, system, tools, image, **input_kwargs)
|
||||
while True:
|
||||
try:
|
||||
|
|
Loading…
Reference in New Issue