diff --git a/src/axolotl/utils/data.py b/src/axolotl/utils/data.py index c944dd27b..99697de32 100644 --- 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