Skip to content

Commit df34cae

Browse files
committed
fix a bug in AbsDataset.py for embedder finetune
1 parent 3f07173 commit df34cae

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

FlagEmbedding/abc/finetune/embedder/AbsDataset.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -196,7 +196,7 @@ def __call__(self, features):
196196
d_collated = self.tokenizer.pad(
197197
passages_inputs,
198198
padding=self.padding,
199-
max_length=self.query_max_len,
199+
max_length=self.passage_max_len,
200200
pad_to_multiple_of=self.pad_to_multiple_of,
201201
return_tensors=self.return_tensors
202202
)
@@ -562,7 +562,7 @@ def __call__(self, features):
562562
q_collated.append(self.tokenizer.pad(
563563
sub_features,
564564
padding=self.padding,
565-
max_length=self.passage_max_len,
565+
max_length=self.query_max_len,
566566
pad_to_multiple_of=self.pad_to_multiple_of,
567567
return_tensors=self.return_tensors,
568568
))

0 commit comments

Comments
 (0)