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

fix 🐛

Browse Source
This commit is contained in:
Sourab Mangrulkar
2023-04-05 02:03:31 +05:30
parent b9433a8208
commit 75131959d1
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/peft/peft_model.py
+1 -1
View File
@@ -352,7 +352,7 @@ class PeftModel(PushToHubMixin, torch.nn.Module):
if (
(getattr(self, "hf_device_map", None) is not None)
and (len(set(self.hf_device_map.values()).intersection({"cpu", "disk"})) > 0)
and len(self.peft_config == 1)
and len(self.peft_config) == 1
):
device_map = kwargs.get("device_map", "auto")
max_memory = kwargs.get("max_memory", None)