fix model card network issue

2023-11-19 23:03:19 +08:00 · 2023-11-19 23:03:19 +08:00 · 211b2db5a8
parent bfb9433165
commit 211b2db5a8
1 changed files with 1 additions and 0 deletions
--- a/src/llmtuner/model/utils.py
+++ b/src/llmtuner/model/utils.py
@ -78,6 +78,7 @@ def get_modelcard_args(
 ) -> Dict[str, Any]:
    return {
        "tasks": "text-generation",
+        "license": "other",
        "finetuned_from": model_args.model_name_or_path,
        "dataset": [dataset.strip() for dataset in data_args.dataset.split(",")],
        "tags": ["llama-factory"] + (["lora"] if finetuning_args.finetuning_type == "lora" else [])