Spaces:
Running
Running
Update app.py
Browse files
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=
|
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()
|