fix: max tokens can only up to 2048 (#2734)

2 years ago · 3a3ca8e6a9
parent 27e678480e
commit 3a3ca8e6a9
1 changed files with 1 additions and 0 deletions
--- a/api/core/model_runtime/model_providers/xinference/llm/llm.py
+++ b/api/core/model_runtime/model_providers/xinference/llm/llm.py
@ -308,6 +308,7 @@ class XinferenceAILargeLanguageModel(LargeLanguageModel):
                type=ParameterType.INT,
                use_template='max_tokens',
                min=1,
+                max=credentials.get('context_length', 2048),
                default=512,
                label=I18nObject(
                    zh_Hans='最大生成长度',