diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2022-10-20 05:18:02 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2022-10-21 06:00:45 +0000 |
| commit | 74088c2a06a975092806362aede22f82716cb011 (patch) | |
| tree | a86f1dcdd6613b6143655cae991bdf7cb24e6762 /modules/hypernetworks/ui.py | |
| parent | 45872181902ada06267e2de601586d512cf5df1a (diff) | |
| download | stable-diffusion-webui-gfx803-74088c2a06a975092806362aede22f82716cb011.tar.gz stable-diffusion-webui-gfx803-74088c2a06a975092806362aede22f82716cb011.tar.bz2 stable-diffusion-webui-gfx803-74088c2a06a975092806362aede22f82716cb011.zip | |
allow float sizes for hypernet's layer_structure
Diffstat (limited to 'modules/hypernetworks/ui.py')
| -rw-r--r-- | modules/hypernetworks/ui.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/modules/hypernetworks/ui.py b/modules/hypernetworks/ui.py index 08f75f15..e0741d08 100644 --- a/modules/hypernetworks/ui.py +++ b/modules/hypernetworks/ui.py @@ -15,7 +15,7 @@ def create_hypernetwork(name, enable_sizes, layer_structure=None, add_layer_norm assert not os.path.exists(fn), f"file {fn} already exists"
if type(layer_structure) == str:
- layer_structure = tuple(map(int, re.sub(r'\D', '', layer_structure)))
+ layer_structure = [float(x.strip()) for x in layer_structure.split(",")]
hypernet = modules.hypernetworks.hypernetwork.Hypernetwork(
name=name,
|