prevent extra modules from being saved/loaded with hypernet
This commit is contained in:
parent
d4ea5f4d86
commit
6a9ea5b41c
|
@ -50,7 +50,7 @@ class Hypernetwork:
|
||||||
self.sd_checkpoint = None
|
self.sd_checkpoint = None
|
||||||
self.sd_checkpoint_name = None
|
self.sd_checkpoint_name = None
|
||||||
|
|
||||||
for size in enable_sizes or [320, 640, 768, 1280]:
|
for size in enable_sizes or []:
|
||||||
self.layers[size] = (HypernetworkModule(size), HypernetworkModule(size))
|
self.layers[size] = (HypernetworkModule(size), HypernetworkModule(size))
|
||||||
|
|
||||||
def weights(self):
|
def weights(self):
|
||||||
|
|
Loading…
Reference in New Issue