Commit 6a9ea5b4 authored by AUTOMATIC's avatar AUTOMATIC

prevent extra modules from being saved/loaded with hypernet

parent d4ea5f4d
......@@ -50,7 +50,7 @@ class Hypernetwork:
self.sd_checkpoint = 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))
def weights(self):
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment