From 16182ea972f8097dd429cf014bc8cc6c8bcce5d2 Mon Sep 17 00:00:00 2001
From: Younes Belkada <49240599+younesbelkada@users.noreply.github.com>
Date: Sun, 29 Jan 2023 11:41:38 +0100
Subject: [PATCH] Apply suggestions from code review

Co-authored-by: Sourab Mangrulkar <13534540+pacman100@users.noreply.github.com>
---
 src/peft/tuners/lora.py  | 3 ---
 src/peft/utils/config.py | 1 +
 2 files changed, 1 insertion(+), 3 deletions(-)

diff --git a/src/peft/tuners/lora.py b/src/peft/tuners/lora.py
index 7a42572..35d6e4f 100644
--- a/src/peft/tuners/lora.py
+++ b/src/peft/tuners/lora.py
@@ -255,9 +255,6 @@ class LoraModel(PushToHubMixin, torch.nn.Module):
         return model
 
 
-    def forward(self, *args, **kwargs):
-        return self.model(*args, **kwargs)
-
     def __getattr__(self, name: str):
         """Forward missing attributes to the wrapped module."""
         try:
diff --git a/src/peft/utils/config.py b/src/peft/utils/config.py
index ad30b32..2a9d264 100644
--- a/src/peft/utils/config.py
+++ b/src/peft/utils/config.py
@@ -124,6 +124,7 @@ class PeftConfigMixin(PushToHubMixin):
         return json_object
 
 
+@dataclass
 class PeftConfig(PeftConfigMixin):
     """
     This is the base configuration class to store the configuration of a :class:`~peft.PeftModel`.