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

Merge pull request #257 from toncho11/main

Browse Source 
Fixing a bug where a wrong parameter name is used for the offload_folder
This commit is contained in:
Sourab Mangrulkar
2023-04-03 22:30:31 +05:30
committed by GitHub
parent 93f1d35cc7 ff9a1edbfd
commit 127a74baa2
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
@@ -164,7 +164,7 @@ class PeftModel(PushToHubMixin, torch.nn.Module):
if getattr(model, "hf_device_map", None) is not None:
device_map = kwargs.get("device_map", "auto")
max_memory = kwargs.get("max_memory", None)
offload_dir = kwargs.get("offload_dir", None)
offload_dir = kwargs.get("offload_folder", None)
offload_index = kwargs.get("offload_index", None)
dispatch_model_kwargs = {}