fixes

2026-09-10 12:20:21 +08:00 · 2023-02-01 17:17:14 +05:30
parent 915a5db0c6
commit fcd213708d
1 changed files with 1 additions and 0 deletions
@@ -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):