From ff9a1edbfd2d405b86d50a2e5299cc1bbd49d887 Mon Sep 17 00:00:00 2001 From: toncho11 Date: Mon, 3 Apr 2023 18:28:11 +0200 Subject: [PATCH] Fixing a bug where a wrong parameter name is used. --- src/peft/peft_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/peft/peft_model.py b/src/peft/peft_model.py index f9573bb..85757b7 100644 --- a/src/peft/peft_model.py +++ b/src/peft/peft_model.py @@ -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 = {}