Fix failing test (#25963)

3a479672 · Lysandre Debut · GitHub · 034bc5d2 · 3a479672
Unverified Commit 3a479672 authored Sep 04, 2023 by Lysandre Debut Committed by GitHub Sep 04, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/transformers/modelcard.py src/transformers/modelcard.py +1 -1

No files found.
--- a/src/transformers/modelcard.py
+++ b/src/transformers/modelcard.py
@@ -459,7 +459,7 @@ 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:
+        if self.finetuned_from is not None and isinstance(self.finetuned_from, str) and len(self.finetuned_from) > 0:
            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)