prevent extra modules from being saved/loaded with hypernet

3 years ago · 6a9ea5b41c
parent d4ea5f4d86
commit 6a9ea5b41c
1 changed files with 1 additions and 1 deletions
--- a/modules/hypernetworks/hypernetwork.py
+++ b/modules/hypernetworks/hypernetwork.py
@ -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):