kris/ForcePilot
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

fix embedding bug

Browse Source
This commit is contained in:
Wenjie Zhang 2024-11-22 00:13:01 +08:00
parent c51c9b152d
commit 5cad89e3d4
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/models/embedding.py
View File

@ -58,7 +58,7 @@ class ZhipuEmbedding:
def predict(self, message): def predict(self, message):
data = [] data = []
batch_size = 40 batch_size = 20
if len(message) > batch_size: if len(message) > batch_size:
global GLOBAL_EMBED_STATE global GLOBAL_EMBED_STATE