xzl12306 commited on
Commit
8f2263e
1 Parent(s): 1b2e2ca

increase timeout

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -166,7 +166,7 @@ def get_response(params):
166
 
167
  stopping_criteria = KeywordsStoppingCriteria(keywords, tokenizer, input_ids)
168
  streamer = TextIteratorStreamer(
169
- tokenizer, skip_prompt=True, skip_special_tokens=True, timeout=15
170
  )
171
 
172
  max_new_tokens = min(
 
166
 
167
  stopping_criteria = KeywordsStoppingCriteria(keywords, tokenizer, input_ids)
168
  streamer = TextIteratorStreamer(
169
+ tokenizer, skip_prompt=True, skip_special_tokens=True, timeout=1500
170
  )
171
 
172
  max_new_tokens = min(