From 9e7b460b2d5251d12c8705f5f045667e9dacca28 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: c53a11b6fd477fffb2030a4e3aeb4a56fd21e7a6 --- 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: