Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 01:04:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #2259 from AI-Casanova/patch-3

Browse Source 
Update lora_diffusers.py for Diffusers 0.21.2
This commit is contained in:
Vladimir Mandic
2023-09-27 09:23:11 -04:00
committed by GitHub
parent ef0e8a5161 7a00e16725
commit 76894461d2
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/lora_diffusers.py
+2 -2
View File
@@ -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