From 7a00e16725be818d4c3f5c5769952e8d9f749283 Mon Sep 17 00:00:00 2001 From: AI-Casanova <54461896+AI-Casanova@users.noreply.github.com> Date: Tue, 26 Sep 2023 18:25:15 -0500 Subject: [PATCH] Update lora_diffusers.py for Diffusers 0.21.2 Caution, NOT backward compatible to 0.20.2 --- modules/lora_diffusers.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/modules/lora_diffusers.py b/modules/lora_diffusers.py index 1e9890f02..c25fdceb3 100644 --- a/modules/lora_diffusers.py +++ b/modules/lora_diffusers.py @@ -249,10 +249,10 @@ class LoRAModule(torch.nn.Module): self.org_module[0].forward = self.org_forward # forward with lora - def forward(self, x): + def forward(self, x, scale = 1.0): if not self.enabled: return self.org_forward(x) - return self.org_forward(x) + self.lora_up(self.lora_down(x)) * self.multiplier * self.scale + return self.org_forward(x, scale = 1.0) + self.lora_up(self.lora_down(x)) * self.multiplier * self.scale def set_network(self, network): self.network = network