From ba8530872094eca773ef2ef30044d9f2926cb3e8 Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Tue, 19 Sep 2023 17:57:07 -0400 Subject: [PATCH] Update src/axolotl/utils/models.py Co-authored-by: NanoCode012 --- src/axolotl/utils/models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/axolotl/utils/models.py b/src/axolotl/utils/models.py index 1719542a1..5a11c1711 100644 --- a/src/axolotl/utils/models.py +++ b/src/axolotl/utils/models.py @@ -472,7 +472,7 @@ def load_adapter(model, cfg, adapter, inference=False): return model, None if hasattr(model, "enable_input_require_grads"): model.enable_input_require_grads() - if adapter in ["ia3"]: + if adapter == "ia3": return load_ia3(model, cfg, inference=inference) if adapter in ["lora", "qlora"]: return load_lora(model, cfg, inference=inference)