prevent extra modules from being saved/loaded with hypernet

rom1win · Oct 11, 2022 · 6a9ea5b · 6a9ea5b
1 parent d4ea5f4
commit 6a9ea5b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/modules/hypernetworks/hypernetwork.py b/modules/hypernetworks/hypernetwork.py
@@ -50,7 +50,7 @@ def __init__(self, name=None, enable_sizes=None):
         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):