From 5cad89e3d4c0bb62bd3a5d634ad55f35754eaaa2 Mon Sep 17 00:00:00 2001
From: Wenjie Zhang <wenjie.zhang@stu.jiangnan.edu.cn>
Date: Fri, 22 Nov 2024 00:13:01 +0800
Subject: [PATCH] fix embedding bug

---
 src/models/embedding.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/models/embedding.py b/src/models/embedding.py
index a124e0ac..7b576abe 100644
--- a/src/models/embedding.py
+++ b/src/models/embedding.py
@@ -58,7 +58,7 @@ class ZhipuEmbedding:
 
     def predict(self, message):
         data = []
-        batch_size = 40
+        batch_size = 20
 
         if len(message) > batch_size:
             global GLOBAL_EMBED_STATE