diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-07-17 06:00:47 +0000 |
|---|---|---|
| committer | AUTOMATIC1111 <16777216c@gmail.com> | 2023-07-17 06:00:47 +0000 |
| commit | 238adeaffb037dedbcefe41e7fd4814a1f17baa2 (patch) | |
| tree | 901557b4f4814effe18d23900317f7dbabf3e162 /extensions-builtin/Lora/network_lokr.py | |
| parent | 46466f09d0b0c14118033dee6af0f876059776d3 (diff) | |
| download | stable-diffusion-webui-gfx803-238adeaffb037dedbcefe41e7fd4814a1f17baa2.tar.gz stable-diffusion-webui-gfx803-238adeaffb037dedbcefe41e7fd4814a1f17baa2.tar.bz2 stable-diffusion-webui-gfx803-238adeaffb037dedbcefe41e7fd4814a1f17baa2.zip | |
support specifying te and unet weights separately
update lora code
support full module
Diffstat (limited to 'extensions-builtin/Lora/network_lokr.py')
| -rw-r--r-- | extensions-builtin/Lora/network_lokr.py | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/extensions-builtin/Lora/network_lokr.py b/extensions-builtin/Lora/network_lokr.py index f1731924..920062e2 100644 --- a/extensions-builtin/Lora/network_lokr.py +++ b/extensions-builtin/Lora/network_lokr.py @@ -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)
|