From 5486ea09e3e7673e802cc97eb97e8da0a4392cfd Mon Sep 17 00:00:00 2001 From: hiyouga Date: Fri, 12 Apr 2024 17:11:59 +0800 Subject: [PATCH] fix model card Former-commit-id: 920e7149bf2b559c9829aa4b11cfb6d00bbb2f9e --- src/llmtuner/train/utils.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/llmtuner/train/utils.py b/src/llmtuner/train/utils.py index cf199633..fef63c39 100644 --- a/src/llmtuner/train/utils.py +++ b/src/llmtuner/train/utils.py @@ -57,9 +57,11 @@ def create_modelcard_and_push( kwargs = { "tasks": "text-generation", "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], } + if data_args.dataset is not None: + kwargs["dataset"] = [dataset.strip() for dataset in data_args.dataset.split(",")] + if not training_args.do_train: pass elif training_args.push_to_hub: