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

---
 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 af17075..38993bf 100644
--- a/src/peft/peft_model.py
+++ b/src/peft/peft_model.py
@@ -76,6 +76,7 @@ class PeftModel(PushToHubMixin, torch.nn.Module):
         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)
+        self.forward = self.base_model.forward
         self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
     def save_pretrained(self, save_directory, **kwargs):