mirror of
https://github.com/hiyouga/LLaMA-Factory.git
synced 2025-12-15 19:30:36 +08:00
tiny fix
This commit is contained in:
@@ -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