From b6c751455e9286290aa9da9d7f9dfe0c0148ee8f Mon Sep 17 00:00:00 2001
From: Sourab Mangrulkar <13534540+pacman100@users.noreply.github.com>
Date: Thu, 6 Apr 2023 19:31:21 +0530
Subject: [PATCH] =?UTF-8?q?=F0=9F=98=85?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/peft/peft_model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/peft/peft_model.py b/src/peft/peft_model.py
index 2b7b286..9bcaf80 100644
--- a/src/peft/peft_model.py
+++ b/src/peft/peft_model.py
@@ -331,7 +331,7 @@ class PeftModel(PushToHubMixin, torch.nn.Module):
             if isinstance(peft_config, PromptLearningConfig) and is_trainable:
                 raise ValueError("Cannot set a prompt learning adapter to trainable when loading pretrained adapter.")
             else:
-                peft_config[adapter_name].inference_mode = not is_trainable
+                peft_config.inference_mode = not is_trainable
             self.add_adapter(adapter_name, peft_config)
 
         # load weights if any