ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-04-30 11:11:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

Upgraded to dev for t2i on diffusers. Minor migrations to make it work.

Browse Source
This commit is contained in:
Jaret Burkett
2023-09-11 14:46:06 -06:00
parent 083cefa78c
commit e8583860ad
7 changed files with 356 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
toolkit/network_mixins.py
View File

@@ -122,11 +122,14 @@ class ToolkitModuleMixin:
return lx * scale
def forward(self: Module, x):
# this may get an additional positional arg or not
def forward(self: Module, x, *args, **kwargs):
# diffusers added scale to resnet.. not sure what it does
if self._multiplier is None:
self.set_multiplier(0.0)
org_forwarded = self.org_forward(x)
org_forwarded = self.org_forward(x, *args, **kwargs)
lora_output = self._call_forward(x)
multiplier = self._multiplier.clone().detach()