Auto modelcard (#11599)
* Autogenerate model cards from the Trainer * ModelCard deprecated * Fix test * Style * Apply suggestions from code review Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com> * Address review comments * Quality * With all metadata * Metadata * Post-merge conflict mess * Data args and all examples * Default license and languages when possible Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
This commit is contained in:
@@ -516,7 +516,14 @@ def main():
|
||||
writer.write(f"{index}\t{item}\n")
|
||||
|
||||
if training_args.push_to_hub:
|
||||
trainer.push_to_hub()
|
||||
kwargs = {"finetuned_from": model_args.model_name_or_path, "tags": "text-classification"}
|
||||
if data_args.task_name is not None:
|
||||
kwargs["language"] = "en"
|
||||
kwargs["dataset_tags"] = "glue"
|
||||
kwargs["dataset_args"] = data_args.task_name
|
||||
kwargs["dataset"] = f"GLUE {data_args.task_name.upper()}"
|
||||
|
||||
trainer.push_to_hub(**kwargs)
|
||||
|
||||
|
||||
def _mp_fn(index):
|
||||
|
||||
Reference in New Issue
Block a user