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 model card

Browse Source 
Former-commit-id: c53a11b6fd477fffb2030a4e3aeb4a56fd21e7a6
This commit is contained in:
hiyouga 2024-04-12 17:11:59 +08:00
parent c9d3cc181a
commit 9e7b460b2d
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: