Skip to content

Commit 476f600

Browse files
authored
remove-redundant-code (#947)
1 parent fd79b31 commit 476f600

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)