wassname/peft
1
0
Fork 0
mirror of https://github.com/wassname/peft.git synced 2026-09-10 12:20:21 +08:00
Code Issues Packages Projects Releases Wiki Activity

fixes

Browse Source
This commit is contained in:
Sourab Mangrulkar
2023-02-01 16:25:42 +05:30
parent d53a631608
commit 915a5db0c6
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/peft/peft_model.py
+5 -1
View File
@@ -96,7 +96,11 @@ class PeftModel(PushToHubMixin, torch.nn.Module):
# save the config
if self.peft_config.base_model_name_or_path is None:
self.peft_config.base_model_name_or_path = self.base_model.__dict__.get("name_or_path", None)
self.peft_config.base_model_name_or_path = (
self.base_model.__dict__.get("name_or_path", None)
if isinstance(self.peft_config, PromptLearningConfig)
else self.base_model.model.__dict__.get("name_or_path", None)
)
self.peft_config.inference_mode = True
self.peft_config.save_pretrained(save_directory)