423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-12-14 19:06:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix model card network issue

Browse Source
This commit is contained in:
hiyouga
2023-11-19 23:03:19 +08:00
parent bfb9433165
commit 211b2db5a8
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/llmtuner/model/utils.py
View File

@@ -78,6 +78,7 @@ def get_modelcard_args(
) -> Dict[str, Any]: ) -> Dict[str, Any]:
return { return {
"tasks": "text-generation", "tasks": "text-generation",
"license": "other",
"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(",")], "dataset": [dataset.strip() for dataset in data_args.dataset.split(",")],
"tags": ["llama-factory"] + (["lora"] if finetuning_args.finetuning_type == "lora" else []) "tags": ["llama-factory"] + (["lora"] if finetuning_args.finetuning_type == "lora" else [])