Spaces:
Running
on
Zero
Running
on
Zero
chiayewken
commited on
Commit
·
0c5ff1b
1
Parent(s):
9ecc669
Bugfix for max_new_tokens
Browse files
app.py
CHANGED
@@ -258,7 +258,11 @@ class QwenModel(EvalModel):
|
|
258 |
skip_special_tokens=True,
|
259 |
)
|
260 |
|
261 |
-
generate_kwargs = dict(
|
|
|
|
|
|
|
|
|
262 |
t = Thread(target=self.model.generate, kwargs=generate_kwargs)
|
263 |
t.start()
|
264 |
|
|
|
258 |
skip_special_tokens=True,
|
259 |
)
|
260 |
|
261 |
+
generate_kwargs = dict(
|
262 |
+
**model_inputs,
|
263 |
+
streamer=streamer,
|
264 |
+
max_new_tokens=self.max_output_tokens,
|
265 |
+
)
|
266 |
t = Thread(target=self.model.generate, kwargs=generate_kwargs)
|
267 |
t.start()
|
268 |
|