Fix: Removed model params except max_token for deepseek r1 in volcengine (#13446)

This commit is contained in:
Fei He 2025-02-10 10:26:26 +08:00 committed by GitHub
parent 75113c26c6
commit 7796984444
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -328,6 +328,19 @@ class VolcengineMaaSLargeLanguageModel(LargeLanguageModel):
""" """
model_config = get_model_config(credentials) model_config = get_model_config(credentials)
if model.startswith("DeepSeek-R1"):
rules = [
ParameterRule(
name="max_tokens",
type=ParameterType.INT,
use_template="max_tokens",
min=1,
max=model_config.properties.max_tokens,
default=512,
label=I18nObject(zh_Hans="最大生成长度", en_US="Max Tokens"),
),
]
else:
rules = [ rules = [
ParameterRule( ParameterRule(
name="temperature", name="temperature",
@ -342,7 +355,11 @@ class VolcengineMaaSLargeLanguageModel(LargeLanguageModel):
label=I18nObject(zh_Hans="Top P", en_US="Top P"), label=I18nObject(zh_Hans="Top P", en_US="Top P"),
), ),
ParameterRule( ParameterRule(
name="top_k", type=ParameterType.INT, min=1, default=1, label=I18nObject(zh_Hans="Top K", en_US="Top K") name="top_k",
type=ParameterType.INT,
min=1,
default=1,
label=I18nObject(zh_Hans="Top K", en_US="Top K"),
), ),
ParameterRule( ParameterRule(
name="presence_penalty", name="presence_penalty",