tiny fix

Former-commit-id: 23961bdf6fdbcde64e7b943f699fdeb4ac024043
2024-08-20 00:10:52 +08:00
parent 388f0a6e05
commit d111a324bc
2 changed files with 4 additions and 4 deletions
--- a/src/llamafactory/train/ppo/trainer.py
+++ b/src/llamafactory/train/ppo/trainer.py
@@ -133,9 +133,9 @@ class CustomPPOTrainer(PPOTrainer, Trainer):
            ref_model=ref_model,
            tokenizer=tokenizer,
            dataset=train_dataset,
+            optimizer=optimizer,
            data_collator=data_collator,
            lr_scheduler=scheduler,
-            optimizer=optimizer,
        )

        self.args = training_args