Model card defaults (#12122)

* [WIP] Model card defaults

* finetuned_from default value

* Add all mappings to the mapping file

* Be more defensive on finetuned_from arg

* Add default task tag

* Separate tags from tasks

* Edge case for dataset

* Apply suggestions from code review

Co-authored-by: Lysandre Debut <lysandre@huggingface.co>

Co-authored-by: Lysandre Debut <lysandre@huggingface.co>
This commit is contained in:
Sylvain Gugger
2021-06-15 16:01:37 -04:00
committed by GitHub
parent 6e7cc5cc51
commit 7d7ceca396
14 changed files with 477 additions and 26 deletions

View File

@@ -430,7 +430,7 @@ def main():
if training_args.push_to_hub:
trainer.push_to_hub(
finetuned_from=model_args.model_name_or_path,
tags="multiple-choice",
tasks="multiple-choice",
dataset_tags="swag",
dataset_args="regular",
dataset="SWAG",