Add `finetuned_from` property in the autogenerated model card (#24528)

* Add finetuned_from tag in the autogenerated model card * Update name

Add `finetuned_from` property in the autogenerated model card (#24528)
* Add finetuned_from tag in the autogenerated model card * Update name
b19c7b5c · Sylvain Gugger · GitHub · ea9caf7a · b19c7b5c
Unverified Commit b19c7b5c authored Jul 04, 2023 by Sylvain Gugger Committed by GitHub Jul 04, 2023
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

src/transformers/modelcard.py src/transformers/modelcard.py +2 -0

No files found.
--- a/src/transformers/modelcard.py
+++ b/src/transformers/modelcard.py
@@ -459,6 +459,8 @@ class TrainingSummary:
        metadata = {}
        metadata = _insert_values_as_list(metadata, "language", self.language)
        metadata = _insert_value(metadata, "license", self.license)
+        if self.finetuned_from is not None:
+            metadata = _insert_value(metadata, "base_model", self.finetuned_from)
        metadata = _insert_values_as_list(metadata, "tags", self.tags)
        metadata = _insert_values_as_list(metadata, "datasets", self.dataset_tags)
        metadata = _insert_values_as_list(metadata, "metrics", list(metric_mapping.keys()))