wassname/peft
1
0
Fork 0
mirror of https://github.com/wassname/peft.git synced 2026-09-24 13:40:15 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update peft_model.py

Browse Source
This commit is contained in:
Sourab Mangrulkar
2023-03-28 19:29:24 +05:30
parent c21afbe868
commit af252b709b
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/peft/peft_model.py
+3 -2
View File
@@ -86,9 +86,10 @@ class PeftModel(PushToHubMixin, torch.nn.Module):
self.active_adapter = adapter_name
if not isinstance(peft_config, PromptLearningConfig):
self.base_model = PEFT_TYPE_TO_MODEL_MAPPING[peft_config.peft_type](
self.base_model, peft_config, adapter_name
self.base_model, self.peft_config, adapter_name
)
self.add_adapter(adapter_name, peft_config)
else:
self.add_adapter(adapter_name, peft_config)
def save_pretrained(self, save_directory, **kwargs):
r"""