zhangshengdong commited on
Commit
8563b4e
1 Parent(s): a41d7ea

Upload chatbot_local.py

Browse files
Files changed (1) hide show
  1. 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=60.0, skip_prompt=True, skip_special_tokens=True)
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: