Skip to content

Commit 06f5df4

Browse files
simonJJJshenzheyu
authored andcommitted
remove-redundant-code (deepspeedai#947)
Signed-off-by: Zheyu SHEN <[email protected]>
1 parent 710d83f commit 06f5df4

File tree

1 file changed

+0
-5
lines changed

1 file changed

+0
-5
lines changed

applications/DeepSpeed-Chat/dschat/utils/data/data_utils.py

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -211,12 +211,7 @@ def create_dataset_split(current_dataset, raw_dataset, train_phase, tokenizer,
211211
padding="max_length",
212212
truncation=True,
213213
return_tensors="pt")
214-
chosen_token["input_ids"] = chosen_token["input_ids"]
215-
chosen_token["attention_mask"] = chosen_token["attention_mask"]
216214
chosen_dataset.append(chosen_token)
217-
218-
reject_token["input_ids"] = reject_token["input_ids"]
219-
reject_token["attention_mask"] = reject_token["attention_mask"]
220215
reject_dataset.append(reject_token)
221216
print(
222217
f'Creating dataset {raw_dataset.dataset_name_clean} for {train_phase=} size={len(chosen_dataset)}'

0 commit comments

Comments
 (0)