Skip to content

D
dify
Unverified Commit e47b5b43 authored by Yeuoly's avatar Yeuoly Committed by GitHub
Browse files
Options

fix: baichuan frequency_penalty (#2446)

parent 21c9d9e2
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 2 deletions
api/core/model_runtime/model_providers/baichuan/llm/baichuan_turbo.py
View file @ e47b5b43
...@@ -128,8 +128,10 @@ class BaichuanModel: ...@@ -128,8 +128,10 @@ class BaichuanModel:
'role': message.role, 'role': message.role,
}) })
# [baichuan] frequency_penalty must be between 1 and 2 # [baichuan] frequency_penalty must be between 1 and 2
if 'frequency_penalty' in parameters:
if parameters['frequency_penalty'] < 1 or parameters['frequency_penalty'] > 2: if parameters['frequency_penalty'] < 1 or parameters['frequency_penalty'] > 2:
parameters['frequency_penalty'] = 1 parameters['frequency_penalty'] = 1
# turbo api accepts flat parameters # turbo api accepts flat parameters
return { return {
'model': self._model_mapping(model), 'model': self._model_mapping(model),
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment