diff --git a/modelscope/pipelines/nlp/llama2_text_generation_pipeline.py b/modelscope/pipelines/nlp/llama2_text_generation_pipeline.py index 3a9d3d44..d366ec9c 100644 --- a/modelscope/pipelines/nlp/llama2_text_generation_pipeline.py +++ b/modelscope/pipelines/nlp/llama2_text_generation_pipeline.py @@ -65,7 +65,7 @@ class Llama2TaskPipeline(TextGenerationPipeline): def forward(self, inputs, - max_length=50, + max_length=2048, do_sample=True, top_p=0.85, temperature=1.0,