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`.