This commit is contained in:
Sourab Mangrulkar
2023-02-15 16:58:33 +05:30
parent af7414a67d
commit c9b225d257
-1
View File
@@ -81,7 +81,6 @@ class PeftModel(PushToHubMixin, torch.nn.Module):
self.modules_to_save = self.peft_config.modules_to_save
_set_trainable(self)
self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
self.forward.__func__.__signature__ = inspect.signature(self.base_model.forward)
def save_pretrained(self, save_directory, **kwargs):
r"""