From fe7f7d67258624f0dcafe54b87fc17d696354270 Mon Sep 17 00:00:00 2001 From: "yuze.zyz" Date: Thu, 3 Aug 2023 14:34:52 +0800 Subject: [PATCH] fix streamer arg Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/13520666 * fix streamer arg --- modelscope/models/nlp/qwen/text_generation.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modelscope/models/nlp/qwen/text_generation.py b/modelscope/models/nlp/qwen/text_generation.py index 46abdf1e..5f9a2fab 100644 --- a/modelscope/models/nlp/qwen/text_generation.py +++ b/modelscope/models/nlp/qwen/text_generation.py @@ -241,6 +241,6 @@ class QWenForTextGeneration(QWenPreTrainedModel): stopping_criteria, prefix_allowed_tokens_fn, synced_gpus, - streamer, + streamer=streamer, **kwargs, )