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, )