Skip to content
4 changes: 0 additions & 4 deletions megatron/training/datasets/sft_dataset.py
Original file line number Diff line number Diff line change
Expand Up @@ -117,10 +117,6 @@ def extend_with_padding(tokens, targets, positions, pad_len):
tokens_list = tokens.tolist()
targets_list = targets.tolist()

# Add EOD, unless it's already present
if tokens_list[-1] != eod:
tokens_list.append(eod)
targets_list.append(eod)

pack_tokens.extend(tokens_list)
pack_targets.extend(targets_list)
Expand Down
Loading