Update src/axolotl/utils/models.py

Co-authored-by: NanoCode012 <kevinvong@rocketmail.com>
2023-09-19 17:57:07 -04:00
parent 998763bade
commit ba85308720
1 changed files with 1 additions and 1 deletions
--- 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)