Spaces:
Runtime error
Runtime error
zhangshengdong
commited on
Commit
•
8563b4e
1
Parent(s):
a41d7ea
Upload chatbot_local.py
Browse files- utils/chatbot_local.py +1 -1
utils/chatbot_local.py
CHANGED
@@ -42,7 +42,7 @@ class ChatBot:
|
|
42 |
|
43 |
streamer = None
|
44 |
if stream:
|
45 |
-
streamer = TextIteratorStreamer(self.tokenizer, timeout=
|
46 |
|
47 |
stopping_criteria = None
|
48 |
if stop:
|
|
|
42 |
|
43 |
streamer = None
|
44 |
if stream:
|
45 |
+
streamer = TextIteratorStreamer(self.tokenizer, timeout=600.0, skip_prompt=True, skip_special_tokens=True)
|
46 |
|
47 |
stopping_criteria = None
|
48 |
if stop:
|