Napuh commited on
Commit
992d57f
·
unverified ·
1 Parent(s): 91a016f

catch ConnectionError when checking dataset from HuggingFace (#743)

Browse files
Files changed (1) hide show
  1. src/axolotl/utils/data.py +1 -1
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