diff --git a/rag/llm/embedding_model.py b/rag/llm/embedding_model.py index fac954da7..221fea198 100644 --- a/rag/llm/embedding_model.py +++ b/rag/llm/embedding_model.py @@ -155,6 +155,7 @@ class QWenEmbed(Base): def encode(self, texts: list, batch_size=10): import dashscope + batch_size = min(batch_size, 4) try: res = [] token_count = 0