fix streamer arg

Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/13520666
* fix streamer arg
This commit is contained in:
yuze.zyz
2023-08-03 14:34:52 +08:00
committed by wenmeng.zwm
parent 0501dfc388
commit fe7f7d6725

View File

@@ -241,6 +241,6 @@ class QWenForTextGeneration(QWenPreTrainedModel):
stopping_criteria,
prefix_allowed_tokens_fn,
synced_gpus,
streamer,
streamer=streamer,
**kwargs,
)