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

fix model card network issue

Browse Source 
Former-commit-id: 211b2db5a8290f6b52f0a076de56fcc2b06671d6
This commit is contained in:
hiyouga 2023-11-19 23:03:19 +08:00
parent 8d82d7e994
commit d1e03512f4
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 [])