kyy/llm_trainer
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix args

Browse Source
This commit is contained in:
hiyouga committed 2023-11-07 16:36:06 +08:00
1 parent 17c64a0579
commit d92f112951
4 files changed
+8 -8
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

src/llmtuner/tuner/dpo/workflow.py
+2 -2
View File
@@ -60,9 +60,9 @@ def run_dpo(
plot_loss(training_args.output_dir, keys=["loss", "eval_loss"])
if training_args.push_to_hub:
trainer.push_to_hub(**generate_model_card())
trainer.push_to_hub(**generate_model_card(model_args, data_args, finetuning_args))
else:
trainer.create_model_card(**generate_model_card())
trainer.create_model_card(**generate_model_card(model_args, data_args, finetuning_args))
# Evaluation
if training_args.do_eval:
src/llmtuner/tuner/pt/workflow.py
+2 -2
View File
@@ -46,9 +46,9 @@ def run_pt(
plot_loss(training_args.output_dir, keys=["loss", "eval_loss"])
if training_args.push_to_hub:
trainer.push_to_hub(**generate_model_card())
trainer.push_to_hub(**generate_model_card(model_args, data_args, finetuning_args))
else:
trainer.create_model_card(**generate_model_card())
trainer.create_model_card(**generate_model_card(model_args, data_args, finetuning_args))
# Evaluation
if training_args.do_eval:
src/llmtuner/tuner/rm/workflow.py
+2 -2
View File
@@ -54,9 +54,9 @@ def run_rm(
plot_loss(training_args.output_dir, keys=["loss", "eval_loss"])
if training_args.push_to_hub:
trainer.push_to_hub(**generate_model_card())
trainer.push_to_hub(**generate_model_card(model_args, data_args, finetuning_args))
else:
trainer.create_model_card(**generate_model_card())
trainer.create_model_card(**generate_model_card(model_args, data_args, finetuning_args))
# Evaluation
if training_args.do_eval:
src/llmtuner/tuner/sft/workflow.py
+2 -2
View File
@@ -73,9 +73,9 @@ def run_sft(
plot_loss(training_args.output_dir, keys=["loss", "eval_loss"])
if training_args.push_to_hub:
trainer.push_to_hub(**generate_model_card())
trainer.push_to_hub(**generate_model_card(model_args, data_args, finetuning_args))
else:
trainer.create_model_card(**generate_model_card())
trainer.create_model_card(**generate_model_card(model_args, data_args, finetuning_args))
# Evaluation
if training_args.do_eval: