We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 3f07173 commit df34caeCopy full SHA for df34cae
1 file changed
FlagEmbedding/abc/finetune/embedder/AbsDataset.py
@@ -196,7 +196,7 @@ def __call__(self, features):
196
d_collated = self.tokenizer.pad(
197
passages_inputs,
198
padding=self.padding,
199
- max_length=self.query_max_len,
+ max_length=self.passage_max_len,
200
pad_to_multiple_of=self.pad_to_multiple_of,
201
return_tensors=self.return_tensors
202
)
@@ -562,7 +562,7 @@ def __call__(self, features):
562
q_collated.append(self.tokenizer.pad(
563
sub_features,
564
565
- max_length=self.passage_max_len,
+ max_length=self.query_max_len,
566
567
return_tensors=self.return_tensors,
568
))
0 commit comments