catch ConnectionError when checking dataset from HuggingFace (#743)

axolotl-ai-cloud · Oct 19, 2023 · 992d57f · 992d57f
1 parent 91a016f
commit 992d57f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/axolotl/utils/data.py b/src/axolotl/utils/data.py
@@ -158,7 +158,7 @@ def for_d_in_datasets(dataset_configs):
                     token=use_auth_token,
                 )
                 ds_from_hub = True
-            except FileNotFoundError:
+            except (FileNotFoundError, ConnectionError):
                 pass
 
             # prefer local dataset, even if hub exists