423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-12-14 19:06:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix args

Browse Source 
Former-commit-id: d92f112951
This commit is contained in:
hiyouga
2023-11-07 16:36:06 +08:00
parent 53fcc531b5
commit b446582bfd
4 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/llmtuner/tuner/dpo/workflow.py
View File

@@ -60,9 +60,9 @@ def run_dpo(
plot_loss(training_args.output_dir, keys=["loss", "eval_loss"]) plot_loss(training_args.output_dir, keys=["loss", "eval_loss"])
if training_args.push_to_hub: 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: else:
trainer.create_model_card(**generate_model_card()) trainer.create_model_card(**generate_model_card(model_args, data_args, finetuning_args))
# Evaluation # Evaluation
if training_args.do_eval: if training_args.do_eval:

4
src/llmtuner/tuner/pt/workflow.py
View File

@@ -46,9 +46,9 @@ def run_pt(
plot_loss(training_args.output_dir, keys=["loss", "eval_loss"]) plot_loss(training_args.output_dir, keys=["loss", "eval_loss"])
if training_args.push_to_hub: 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: else:
trainer.create_model_card(**generate_model_card()) trainer.create_model_card(**generate_model_card(model_args, data_args, finetuning_args))
# Evaluation # Evaluation
if training_args.do_eval: if training_args.do_eval:

4
src/llmtuner/tuner/rm/workflow.py
View File

@@ -54,9 +54,9 @@ def run_rm(
plot_loss(training_args.output_dir, keys=["loss", "eval_loss"]) plot_loss(training_args.output_dir, keys=["loss", "eval_loss"])
if training_args.push_to_hub: 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: else:
trainer.create_model_card(**generate_model_card()) trainer.create_model_card(**generate_model_card(model_args, data_args, finetuning_args))
# Evaluation # Evaluation
if training_args.do_eval: if training_args.do_eval:

4
src/llmtuner/tuner/sft/workflow.py
View File

@@ -73,9 +73,9 @@ def run_sft(
plot_loss(training_args.output_dir, keys=["loss", "eval_loss"]) plot_loss(training_args.output_dir, keys=["loss", "eval_loss"])
if training_args.push_to_hub: 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: else:
trainer.create_model_card(**generate_model_card()) trainer.create_model_card(**generate_model_card(model_args, data_args, finetuning_args))
# Evaluation # Evaluation
if training_args.do_eval: if training_args.do_eval: