Napuh
commited on
catch ConnectionError when checking dataset from HuggingFace (#743)
Browse files
src/axolotl/utils/data.py
CHANGED
@@ -158,7 +158,7 @@ def load_tokenized_prepared_datasets(
|
|
158 |
token=use_auth_token,
|
159 |
)
|
160 |
ds_from_hub = True
|
161 |
-
except FileNotFoundError:
|
162 |
pass
|
163 |
|
164 |
# prefer local dataset, even if hub exists
|
|
|
158 |
token=use_auth_token,
|
159 |
)
|
160 |
ds_from_hub = True
|
161 |
+
except (FileNotFoundError, ConnectionError):
|
162 |
pass
|
163 |
|
164 |
# prefer local dataset, even if hub exists
|