hsuwill000 commited on
Commit
f123151
·
verified ·
1 Parent(s): afcd897

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -24,7 +24,7 @@ def generate_stream(prompt):
24
 
25
  def worker():
26
  # 在背景 thread 中做推論
27
- pipe.generate([prompt], streamer=streamer, max_new_tokens=32768)
28
  q.put(None) # 結束符號
29
 
30
  threading.Thread(target=worker).start()
 
24
 
25
  def worker():
26
  # 在背景 thread 中做推論
27
+ pipe.generate([prompt], streamer=streamer, max_new_tokens=4096)
28
  q.put(None) # 結束符號
29
 
30
  threading.Thread(target=worker).start()