fix embedding bug

2024-11-22 00:13:01 +08:00 · 2024-11-22 00:13:01 +08:00 · 5cad89e3d4
commit 5cad89e3d4
parent c51c9b152d
1 changed files with 1 additions and 1 deletions
--- 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