Skip to content

Commit f42abcf

Browse files
committed
Fix conflict and add check for batch_size
1 parent 002add0 commit f42abcf

File tree

2 files changed

+1
-1
lines changed

2 files changed

+1
-1
lines changed

tools/who_what_benchmark/whowhatbench/embeddings_evaluator.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -182,6 +182,7 @@ def default_gen_answer(model, tokenizer, passages, **kwargs):
182182
'normalize': self.normalize}
183183

184184
batch_size = self.batch_size or len(data[0])
185+
batch_size = min(batch_size, len(data[0]))
185186
data_input = data[0][:batch_size]
186187
result = gen_answer_fn(model, self.tokenizer, data_input, **kwargs)
187188

tools/who_what_benchmark/whowhatbench/wwb.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -523,7 +523,6 @@ def genai_gen_visual_text(model, prompt, image, processor, tokenizer, max_new_to
523523

524524
def genai_gen_embedding(model, tokenizer, passages, **kwargs):
525525
embeddings = model.embed_documents(passages)
526-
527526
return embeddings
528527

529528

0 commit comments

Comments
 (0)