Skip to content

Commit 1c4b295

Browse files
committed
revert modif online dpo
1 parent adcb80b commit 1c4b295

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

trl/trainer/online_dpo_trainer.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -68,6 +68,7 @@
6868
disable_dropout_in_model,
6969
empty_cache,
7070
ensure_master_addr_port,
71+
pad,
7172
prepare_deepspeed,
7273
truncate_right,
7374
)

0 commit comments

Comments
 (0)