Spaces:
Runtime error
Runtime error
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 |
|