From e47b5b43b8c372e22f0ade1cbf42325de89817ea Mon Sep 17 00:00:00 2001 From: Yeuoly <45712896+Yeuoly@users.noreply.github.com> Date: Wed, 14 Feb 2024 20:11:41 +0800 Subject: [PATCH] fix: baichuan frequency_penalty (#2446) --- .../model_providers/baichuan/llm/baichuan_turbo.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/api/core/model_runtime/model_providers/baichuan/llm/baichuan_turbo.py b/api/core/model_runtime/model_providers/baichuan/llm/baichuan_turbo.py index ae73c1735..639f6a21c 100644 --- a/api/core/model_runtime/model_providers/baichuan/llm/baichuan_turbo.py +++ b/api/core/model_runtime/model_providers/baichuan/llm/baichuan_turbo.py @@ -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),