catch ConnectionError when checking dataset from HuggingFace (#743)

2023-10-19 04:11:54 +02:00
parent 91a016f410
commit 992d57f20a
1 changed files with 1 additions and 1 deletions
--- a/src/axolotl/utils/data.py
+++ b/src/axolotl/utils/data.py
@@ -158,7 +158,7 @@ def load_tokenized_prepared_datasets(
                    token=use_auth_token,
                )
                ds_from_hub = True
-            except FileNotFoundError:
+            except (FileNotFoundError, ConnectionError):
                pass

            # prefer local dataset, even if hub exists