From 88b4d69278aa33bf2d97d3462255bebcbf72f34d Mon Sep 17 00:00:00 2001
From: xielong <xielong.me@gmail.com>
Date: Tue, 28 May 2024 16:37:44 +0800
Subject: [PATCH] fix: Correct context size for banchuan2-53b and
 banchuan2-turbo (#4721)

---
 .../model_providers/baichuan/llm/baichuan2-53b.yaml             | 2 +-
 .../model_providers/baichuan/llm/baichuan2-turbo.yaml           | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/api/core/model_runtime/model_providers/baichuan/llm/baichuan2-53b.yaml b/api/core/model_runtime/model_providers/baichuan/llm/baichuan2-53b.yaml
index e7cf059ea..04849500d 100644
--- a/api/core/model_runtime/model_providers/baichuan/llm/baichuan2-53b.yaml
+++ b/api/core/model_runtime/model_providers/baichuan/llm/baichuan2-53b.yaml
@@ -6,7 +6,7 @@ features:
   - agent-thought
 model_properties:
   mode: chat
-  context_size: 4000
+  context_size: 32000
 parameter_rules:
   - name: temperature
     use_template: temperature
diff --git a/api/core/model_runtime/model_providers/baichuan/llm/baichuan2-turbo.yaml b/api/core/model_runtime/model_providers/baichuan/llm/baichuan2-turbo.yaml
index fb5d73b06..f91329c77 100644
--- a/api/core/model_runtime/model_providers/baichuan/llm/baichuan2-turbo.yaml
+++ b/api/core/model_runtime/model_providers/baichuan/llm/baichuan2-turbo.yaml
@@ -6,7 +6,7 @@ features:
   - agent-thought
 model_properties:
   mode: chat
-  context_size: 192000
+  context_size: 32000
 parameter_rules:
   - name: temperature
     use_template: temperature