From 96ca100e34e1d0207a517d34ccc9a73fb4abf69e Mon Sep 17 00:00:00 2001 From: Sourab Mangrulkar <13534540+pacman100@users.noreply.github.com> Date: Tue, 4 Apr 2023 18:03:32 +0530 Subject: [PATCH] Update lora.py --- src/peft/tuners/lora.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/peft/tuners/lora.py b/src/peft/tuners/lora.py index 0b96a77..aab5cdf 100644 --- a/src/peft/tuners/lora.py +++ b/src/peft/tuners/lora.py @@ -166,7 +166,7 @@ class LoraModel(torch.nn.Module): "fan_in_fan_out": lora_config.fan_in_fan_out, "merge_weights": (lora_config.merge_weights or lora_config.inference_mode) and not is_hf_device_map_available, - "init_lora_weights": self.peft_config.init_lora_weights, + "init_lora_weights": lora_config.init_lora_weights, } key_list = [key for key, _ in self.model.named_modules()] for key in key_list: