Skip to content

Commit 05005a9

Browse files
committed
fix DecoderOnlyEmbedderICLSameDatasetTrainDataset category index error
1 parent 9abcfc8 commit 05005a9

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

  • FlagEmbedding/finetune/embedder/decoder_only/icl

FlagEmbedding/finetune/embedder/decoder_only/icl/dataset.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -113,8 +113,8 @@ def _create_batch_data(self, batch_raw_data):
113113
add_special_tokens=False,
114114
)['input_ids']
115115
)
116-
for i in range(len(tmp_passages)):
117-
tmp_passages[i] += self.args.icl_suffix_str
116+
for j in range(len(tmp_passages)):
117+
tmp_passages[j] += self.args.icl_suffix_str
118118
else:
119119
if self.args.passage_instruction_for_retrieval is not None:
120120
tmp_passages = [

0 commit comments

Comments
 (0)