Browse Source

fix: baichuan frequency_penalty (#2446)

tags/0.5.6
Yeuoly 1 year ago
parent
commit
e47b5b43b8
No account linked to committer's email address

+ 4
- 2
api/core/model_runtime/model_providers/baichuan/llm/baichuan_turbo.py View File

@@ -128,8 +128,10 @@ class BaichuanModel:
'role': message.role,
})
# [baichuan] frequency_penalty must be between 1 and 2
if parameters['frequency_penalty'] < 1 or parameters['frequency_penalty'] > 2:
parameters['frequency_penalty'] = 1
if 'frequency_penalty' in parameters:
if parameters['frequency_penalty'] < 1 or parameters['frequency_penalty'] > 2:
parameters['frequency_penalty'] = 1

# turbo api accepts flat parameters
return {
'model': self._model_mapping(model),

Loading…
Cancel
Save