Fix failing test

2023-06-09 00:35:03 +09:00
parent babf0fdb71
commit bfd27ba55e
1 changed files with 1 additions and 1 deletions
--- a/src/axolotl/utils/validation.py
+++ b/src/axolotl/utils/validation.py
@@ -54,7 +54,7 @@ def validate_config(cfg):
            "Require cfg.hf_use_auth_token to be True for push_dataset_to_hub"
        )

-    if "falcon" in cfg.base_model.lower() and cfg.fsdp:
+    if (cfg.base_model and "falcon" in cfg.base_model.lower()) and cfg.fsdp:
        raise ValueError("FSDP is not supported for falcon models")

    # TODO