Fix requests connection error during modelcard creation (huggingface#…

…26518) fix requests connection error Co-authored-by: Jan Philipp Harries <[email protected]>
blbadger · Nov 8, 2023 · 883676b · 883676b
1 parent 1eee66b
commit 883676b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/transformers/modelcard.py b/src/transformers/modelcard.py
@@ -387,7 +387,7 @@ def __post_init__(self):
                 for tag in info.tags:
                     if tag.startswith("license:"):
                         self.license = tag[8:]
-            except (requests.exceptions.HTTPError, HFValidationError):
+            except (requests.exceptions.HTTPError, requests.exceptions.ConnectionError, HFValidationError):
                 pass
 
     def create_model_index(self, metric_mapping):