From 95907aed31dddd4ae26dcb01a3161a65fbe0db1d Mon Sep 17 00:00:00 2001
From: mushenL <125954878+mushenL@users.noreply.github.com>
Date: Tue, 4 Jul 2023 18:42:09 +0800
Subject: [PATCH] Modify the parameter passing of the text_generation_pipeline
 class (#355)

---
 modelscope/pipelines/nlp/text_generation_pipeline.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/modelscope/pipelines/nlp/text_generation_pipeline.py b/modelscope/pipelines/nlp/text_generation_pipeline.py
index cb9c89db..9e25ea95 100644
--- a/modelscope/pipelines/nlp/text_generation_pipeline.py
+++ b/modelscope/pipelines/nlp/text_generation_pipeline.py
@@ -66,7 +66,8 @@ class TextGenerationPipeline(Pipeline, PipelineStreamingOutputMixin):
             device=device,
             auto_collate=auto_collate,
             compile=kwargs.pop('compile', False),
-            compile_options=kwargs.pop('compile_options', {}))
+            compile_options=kwargs.pop('compile_options', {}),
+            **kwargs)
 
         assert isinstance(self.model, Model), \
             f'please check whether model config exists in {ModelFile.CONFIGURATION}'