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

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

diff --git a/src/peft/peft_model.py b/src/peft/peft_model.py
index b54a148..71b9ef2 100644
--- a/src/peft/peft_model.py
+++ b/src/peft/peft_model.py
@@ -85,6 +85,7 @@ class PeftModel(PushToHubMixin, torch.nn.Module):
         self.peft_config = {}
         self.active_adapter = adapter_name
         if not isinstance(peft_config, PromptLearningConfig):
+            self.peft_config[adapter_name] = peft_config
             self.base_model = PEFT_TYPE_TO_MODEL_MAPPING[peft_config.peft_type](
                 self.base_model, self.peft_config, adapter_name
             )