fix model card network issue
This commit is contained in:
1 parent
bfb9433165
commit
211b2db5a8
1 file changed
+1
@@ -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 [])
|
||||||
|
|||||||
Reference in new issue
Block a user