Spaces:

mPLUG
/

TinyChart-3B

Running on Zero

xzl12306 commited on Apr 28, 2024

Commit

8f2263e

1 Parent(s): 1b2e2ca

increase timeout

Files changed (1) hide show

app.py CHANGED Viewed

@@ -166,7 +166,7 @@ def get_response(params):
     stopping_criteria = KeywordsStoppingCriteria(keywords, tokenizer, input_ids)
     streamer = TextIteratorStreamer(
-        tokenizer, skip_prompt=True, skip_special_tokens=True, timeout=15
     )
     max_new_tokens = min(

     stopping_criteria = KeywordsStoppingCriteria(keywords, tokenizer, input_ids)
     streamer = TextIteratorStreamer(
+        tokenizer, skip_prompt=True, skip_special_tokens=True, timeout=1500
     )
     max_new_tokens = min(