wassname/peft
1
0
Fork 0
mirror of https://github.com/wassname/peft.git synced 2026-09-09 11:28:32 +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:40:06 +05:30
parent 090d074399
commit 7c8ee5814a
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/peft/peft_model.py
+2 -2
View File
@@ -324,8 +324,8 @@ class PeftModel(PushToHubMixin, torch.nn.Module):
self.add_adapter(adapter_name, peft_config)
# load weights if any
if kwargs.get("subfolder", None) is not None:
path = os.path.join(model_id, kwargs["subfolder"])
path = os.path.join(model_id, kwargs["subfolder"]) if kwargs.get("subfolder", None) is not None else model_id
if os.path.exists(os.path.join(path, WEIGHTS_NAME)):
filename = os.path.join(path, WEIGHTS_NAME)
else: