This commit is contained in:
hiyouga
2024-03-26 17:26:14 +08:00
parent 7ea1a1f5b3
commit 511f675402
7 changed files with 36 additions and 31 deletions

View File

@@ -45,13 +45,10 @@ def run_dpo(
# Initialize our Trainer
trainer = CustomDPOTrainer(
beta=finetuning_args.dpo_beta,
loss_type=finetuning_args.dpo_loss,
ftx_gamma=finetuning_args.dpo_ftx,
finetuning_args=finetuning_args,
model=model,
ref_model=ref_model,
args=training_args,
finetuning_args=finetuning_args,
tokenizer=tokenizer,
data_collator=data_collator,
callbacks=callbacks,