From bfd27ba55efb181cca7d9a86308bfd53d6c54272 Mon Sep 17 00:00:00 2001 From: NanoCode012 Date: Fri, 9 Jun 2023 00:35:03 +0900 Subject: [PATCH] Fix failing test --- src/axolotl/utils/validation.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/axolotl/utils/validation.py b/src/axolotl/utils/validation.py index 367178719..04ffc4c1b 100644 --- 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