fix bugs

huggingface · Sep 5, 2023 · 19364f4 · 19364f4
1 parent 4eb412f
commit 19364f4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/peft/tuners/lora/model.py b/src/peft/tuners/lora/model.py
@@ -671,7 +671,7 @@ def unload(self):
     def forward(self, *args: Any, **kwargs: Any):
         adapter_names = kwargs.pop("adapter_names", None)
         if adapter_names is None:
-            return super().forward(*args, **kwargs)
+            return self.model.forward(*args, **kwargs)
 
         if self.training:
             raise ValueError("Multiple LoRAs in the same batch isn't supported during training")
@@ -694,7 +694,7 @@ def forward(self, *args: Any, **kwargs: Any):
     def generate(self, **kwargs: Any):
         adapter_names = kwargs.pop("adapter_names", None)
         if adapter_names is None:
-            return super().forward(**kwargs)
+            return self.model.generate(**kwargs)
 
         if self.training:
             raise ValueError("Multiple LoRAs in the same batch isn't supported during training")