From af252b709bd8f69dc47c10efb0c9798ec7f639de Mon Sep 17 00:00:00 2001
From: Sourab Mangrulkar <13534540+pacman100@users.noreply.github.com>
Date: Tue, 28 Mar 2023 19:29:24 +0530
Subject: [PATCH] Update peft_model.py

---
 src/peft/peft_model.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/peft/peft_model.py b/src/peft/peft_model.py
index cbc8a89..b54a148 100644
--- a/src/peft/peft_model.py
+++ b/src/peft/peft_model.py
@@ -86,9 +86,10 @@ class PeftModel(PushToHubMixin, torch.nn.Module):
         self.active_adapter = adapter_name
         if not isinstance(peft_config, PromptLearningConfig):
             self.base_model = PEFT_TYPE_TO_MODEL_MAPPING[peft_config.peft_type](
-                self.base_model, peft_config, adapter_name
+                self.base_model, self.peft_config, adapter_name
             )
-        self.add_adapter(adapter_name, peft_config)
+        else:
+            self.add_adapter(adapter_name, peft_config)
 
     def save_pretrained(self, save_directory, **kwargs):
         r"""