Unverified Commit b19c7b5c authored by Sylvain Gugger's avatar Sylvain Gugger Committed by GitHub
Browse files

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

* Add finetuned_from tag in the autogenerated model card

* Update name
parent ea9caf7a
...@@ -459,6 +459,8 @@ class TrainingSummary: ...@@ -459,6 +459,8 @@ class TrainingSummary:
metadata = {} metadata = {}
metadata = _insert_values_as_list(metadata, "language", self.language) metadata = _insert_values_as_list(metadata, "language", self.language)
metadata = _insert_value(metadata, "license", self.license) 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, "tags", self.tags)
metadata = _insert_values_as_list(metadata, "datasets", self.dataset_tags) metadata = _insert_values_as_list(metadata, "datasets", self.dataset_tags)
metadata = _insert_values_as_list(metadata, "metrics", list(metric_mapping.keys())) metadata = _insert_values_as_list(metadata, "metrics", list(metric_mapping.keys()))
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment