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
|