fix: baichuan frequency_penalty (#2446)

e47b5b43 · Yeuoly · GitHub · 21c9d9e2 · e47b5b43
Unverified Commit e47b5b43 authored Feb 14, 2024 by Yeuoly Committed by GitHub Feb 14, 2024
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 2 deletions

baichuan_turbo.py ...el_runtime/model_providers/baichuan/llm/baichuan_turbo.py +4 -2

No files found.
--- 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 '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),