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