diff --git a/FlagEmbedding/finetune/embedder/decoder_only/icl/dataset.py b/FlagEmbedding/finetune/embedder/decoder_only/icl/dataset.py index 802aa39d..43cc8013 100644 --- a/FlagEmbedding/finetune/embedder/decoder_only/icl/dataset.py +++ b/FlagEmbedding/finetune/embedder/decoder_only/icl/dataset.py @@ -113,8 +113,8 @@ def _create_batch_data(self, batch_raw_data): add_special_tokens=False, )['input_ids'] ) - for i in range(len(tmp_passages)): - tmp_passages[i] += self.args.icl_suffix_str + for j in range(len(tmp_passages)): + tmp_passages[j] += self.args.icl_suffix_str else: if self.args.passage_instruction_for_retrieval is not None: tmp_passages = [