Catch HFValidationError in TrainingSummary (huggingface#19252)

* Catch HfValidationError in TrainingSummary Co-authored-by: ydshieh <[email protected]>
impira · Sep 30, 2022 · 4fd32a1 · 4fd32a1
1 parent f3d2f7a
commit 4fd32a1
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/transformers/modelcard.py b/src/transformers/modelcard.py
@@ -26,6 +26,7 @@
 import requests
 import yaml
 from huggingface_hub import model_info
+from huggingface_hub.utils import HFValidationError
 
 from . import __version__
 from .models.auto.modeling_auto import (
@@ -378,7 +379,7 @@ def __post_init__(self):
                 for tag in info.tags:
                     if tag.startswith("license:"):
                         self.license = tag[8:]
-            except requests.exceptions.HTTPError:
+            except (requests.exceptions.HTTPError, HFValidationError):
                 pass
 
     def create_model_index(self, metric_mapping):