mirror of
https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git
synced 2025-08-15 14:15:52 +08:00
Fix: Removed model params except max_token for deepseek r1 in volcengine (#13446)
This commit is contained in:
parent
75113c26c6
commit
7796984444
@ -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",
|
||||||
|
Loading…
x
Reference in New Issue
Block a user