hoelee/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-08-08 13:19:54 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

support specifying te and unet weights separately

Browse Source 
update lora code
support full module
This commit is contained in:
AUTOMATIC1111
2023-07-17 09:00:47 +03:00
parent 46466f09d0
commit 238adeaffb
10 changed files with 151 additions and 78 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
extensions-builtin/Lora/network_lokr.py
View File

@@ -2,7 +2,6 @@ import torch
import lyco_helpers
import network
import network_lyco
class ModuleTypeLokr(network.ModuleType):
@@ -22,7 +21,7 @@ def make_kron(orig_shape, w1, w2):
return torch.kron(w1, w2).reshape(orig_shape)
class NetworkModuleLokr(network_lyco.NetworkModuleLyco):
class NetworkModuleLokr(network.NetworkModule):
def __init__(self, net: network.Network, weights: network.NetworkWeights):
super().__init__(net, weights)