From b9433a82082f7bb56504d5e0d04558f69d5e929f Mon Sep 17 00:00:00 2001 From: Sourab Mangrulkar <13534540+pacman100@users.noreply.github.com> Date: Wed, 5 Apr 2023 01:39:51 +0530 Subject: [PATCH] =?UTF-8?q?=F0=9F=98=85?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/peft/utils/other.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/peft/utils/other.py b/src/peft/utils/other.py index 7cd285f..53e2ee8 100644 --- a/src/peft/utils/other.py +++ b/src/peft/utils/other.py @@ -139,7 +139,7 @@ def _set_trainable(model, adapter_name): for key in key_list: target_module_found = any(key.endswith(target_key) for target_key in model.modules_to_save) if target_module_found: - parent, target, target_name = _get_submodules(key) + parent, target, target_name = _get_submodules(model, key) if isinstance(target, ModulesToSaveWrapper): target.update(adapter_name) else: