Nanobit commited on
Commit
bfd27ba
1 Parent(s): babf0fd

Fix failing test

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