tiny fix
This commit is contained in:
1 parent
15be296347
commit
c8b4c7fee5
2 files changed
+4
-4
No files matched your search
@@ -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
|
||||
|
||||
Reference in new issue
Block a user