diff options
author | AUTOMATIC <16777216c@gmail.com> | 2022-10-11 18:04:47 +0300 |
---|---|---|
committer | AUTOMATIC <16777216c@gmail.com> | 2022-10-11 18:04:47 +0300 |
commit | d682444ecc99319fbd2b142a12727501e2884ba7 (patch) | |
tree | 0b4cb662da210c35c513e95b2786d2eb4082d6df /modules/hypernetworks/hypernetwork.py | |
parent | 5ba23cb41f28f5856a7f64cb0d95e1e94dce90af (diff) |
add option to select hypernetwork modules when creating
Diffstat (limited to 'modules/hypernetworks/hypernetwork.py')
-rw-r--r-- | modules/hypernetworks/hypernetwork.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/modules/hypernetworks/hypernetwork.py b/modules/hypernetworks/hypernetwork.py index aa701bda..b081f14e 100644 --- a/modules/hypernetworks/hypernetwork.py +++ b/modules/hypernetworks/hypernetwork.py @@ -42,7 +42,7 @@ class Hypernetwork: filename = None
name = None
- def __init__(self, name=None):
+ def __init__(self, name=None, enable_sizes=None):
self.filename = None
self.name = name
self.layers = {}
@@ -50,7 +50,7 @@ class Hypernetwork: self.sd_checkpoint = None
self.sd_checkpoint_name = None
- for size in [320, 640, 768, 1280]:
+ for size in enable_sizes or [320, 640, 768, 1280]:
self.layers[size] = (HypernetworkModule(size), HypernetworkModule(size))
def weights(self):
|