From d53a631608e1e3de3eaac288d5c30e7ed14e4abb Mon Sep 17 00:00:00 2001
From: Sourab Mangrulkar <13534540+pacman100@users.noreply.github.com>
Date: Wed, 1 Feb 2023 15:59:24 +0530
Subject: [PATCH] fixes

---
 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 89f0e0d..eba63a6 100644
--- a/src/peft/peft_model.py
+++ b/src/peft/peft_model.py
@@ -75,7 +75,7 @@ class PeftModel(PushToHubMixin, torch.nn.Module):
             self.base_model = LoraModel(peft_config, model)
         if getattr(self.peft_config, "modules_to_save", None) is not None:
             self.modules_to_save = self.peft_config.modules_to_save
-            _set_trainable(self.base_model)
+            _set_trainable(self)
         self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
     def save_pretrained(self, save_directory, **kwargs):