423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-10-16 00:28:10 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix model card network issue

Browse Source 
Former-commit-id: 36155cd1893bea036f15c648c06b0047c02dfb4f
This commit is contained in:
hiyouga 2023-11-19 23:03:19 +08:00
parent 9f364d3880
commit d2ff09a404
1 changed files with 1 additions and 0 deletions
Show Stats 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 [])