allow float sizes for hypernet's layer_structure

2022-10-20 08:18:02 +03:00 · 2022-10-20 08:18:02 +03:00 · 74088c2a06
parent 4587218190
commit 74088c2a06
1 changed files with 1 additions and 1 deletions
--- 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,