richardr1126 commited on
Commit
9bf6dd2
·
1 Parent(s): 097b91f
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -139,7 +139,7 @@ def generate(input_message: str, db_info="", temperature=0.2, top_p=0.9, top_k=0
139
 
140
  input_ids = tok(messages, return_tensors="pt").input_ids
141
  input_ids = input_ids.to(m.device)
142
- streamer = TextIteratorStreamer(tok, timeout=100.0, skip_prompt=True, skip_special_tokens=True)
143
  generate_kwargs = dict(
144
  input_ids=input_ids,
145
  max_new_tokens=max_new_tokens,
 
139
 
140
  input_ids = tok(messages, return_tensors="pt").input_ids
141
  input_ids = input_ids.to(m.device)
142
+ streamer = TextIteratorStreamer(tok, timeout=1000.0, skip_prompt=True, skip_special_tokens=True)
143
  generate_kwargs = dict(
144
  input_ids=input_ids,
145
  max_new_tokens=max_new_tokens,