Kevin Hu commited on
Commit
ef1a16c
·
1 Parent(s): dba7799

fix batch size error for qianwen embedding (#2431)

Browse files

### What problem does this PR solve?

#2402

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)

Files changed (1) hide show
  1. rag/llm/embedding_model.py +1 -0
rag/llm/embedding_model.py CHANGED
@@ -155,6 +155,7 @@ class QWenEmbed(Base):
155
 
156
  def encode(self, texts: list, batch_size=10):
157
  import dashscope
 
158
  try:
159
  res = []
160
  token_count = 0
 
155
 
156
  def encode(self, texts: list, batch_size=10):
157
  import dashscope
158
+ batch_size = min(batch_size, 4)
159
  try:
160
  res = []
161
  token_count = 0