add support for empty list as input to create_model_card (#36042)

handle cases where it is list
This commit is contained in:
ROZBEH 2025-02-05 04:29:17 -08:00 committed by MekkCyber
parent c21f13fe21
commit c14c2cc006

View file

@ -489,7 +489,7 @@ class TrainingSummary:
f" [{self.finetuned_from}](https://huggingface.co/{self.finetuned_from}) on "
)
if self.dataset is None:
if self.dataset is None or (isinstance(self.dataset, list) and len(self.dataset) == 0):
model_card += "an unknown dataset."
else:
if isinstance(self.dataset, str):