Fix failing test
Browse files
src/axolotl/utils/validation.py
CHANGED
@@ -54,7 +54,7 @@ def validate_config(cfg):
|
|
54 |
"Require cfg.hf_use_auth_token to be True for push_dataset_to_hub"
|
55 |
)
|
56 |
|
57 |
-
if "falcon" in cfg.base_model.lower() and cfg.fsdp:
|
58 |
raise ValueError("FSDP is not supported for falcon models")
|
59 |
|
60 |
# TODO
|
|
|
54 |
"Require cfg.hf_use_auth_token to be True for push_dataset_to_hub"
|
55 |
)
|
56 |
|
57 |
+
if (cfg.base_model and "falcon" in cfg.base_model.lower()) and cfg.fsdp:
|
58 |
raise ValueError("FSDP is not supported for falcon models")
|
59 |
|
60 |
# TODO
|