Skip to content

Commit 75e6269

Browse files
committed
fix grpo compat transformers 4.51.3 (#5354)
1 parent d8830ed commit 75e6269

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

swift/trainers/rlhf_trainer/grpo_trainer.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,7 @@
2626
from transformers.trainer import Trainer
2727
from trl import GRPOTrainer as HFGRPOTrainer
2828
from trl.models import prepare_deepspeed
29+
from trl.trainer import grpo_trainer
2930
from trl.trainer.callbacks import SyncRefModelCallback
3031
from trl.trainer.grpo_trainer import RepeatSampler, nanmax, nanmin, nanstd
3132
from trl.trainer.utils import selective_log_softmax
@@ -53,6 +54,7 @@
5354

5455
del HFGRPOTrainer.__init__
5556
del HFGRPOTrainer.log
57+
grpo_trainer.seed_worker = seed_worker # fix transformers 4.51.3
5658

5759
logger = get_logger()
5860
if is_wandb_available():

0 commit comments

Comments
 (0)