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

fix model card

Browse Source 
Former-commit-id: 920e7149bf2b559c9829aa4b11cfb6d00bbb2f9e
This commit is contained in:
hiyouga 2024-04-12 17:11:59 +08:00
parent 31bbbb6d13
commit 5486ea09e3
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/llmtuner/train/utils.py
View File

@ -57,9 +57,11 @@ def create_modelcard_and_push(
kwargs = { kwargs = {
"tasks": "text-generation", "tasks": "text-generation",
"finetuned_from": model_args.model_name_or_path, "finetuned_from": model_args.model_name_or_path,
"dataset": [dataset.strip() for dataset in data_args.dataset.split(",")],
"tags": ["llama-factory", finetuning_args.finetuning_type], "tags": ["llama-factory", finetuning_args.finetuning_type],
} }
if data_args.dataset is not None:
kwargs["dataset"] = [dataset.strip() for dataset in data_args.dataset.split(",")]
if not training_args.do_train: if not training_args.do_train:
pass pass
elif training_args.push_to_hub: elif training_args.push_to_hub: