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

fix args

Browse Source 
Former-commit-id: d92f112951a8d8b28b180c3e2f504a094a9885dd
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
Show Stats 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: