aboutsummaryrefslogtreecommitdiff
path: root/modules/hypernetworks
diff options
context:
space:
mode:
authorAngelBottomless <35677394+aria1th@users.noreply.github.com>2022-10-27 15:40:24 +0900
committerGitHub <noreply@github.com>2022-10-27 15:40:24 +0900
commit462e6ba6675bd14c0f82e465423a0eedfff82372 (patch)
tree905a4025eb5105b2a42525c0fa3cbd69036b2ebe /modules/hypernetworks
parent737eb28faca8be2bb996ee0930ec77d1f7ebd939 (diff)
Disable unavailable or duplicate options
Diffstat (limited to 'modules/hypernetworks')
-rw-r--r--modules/hypernetworks/ui.py3
1 files changed, 2 insertions, 1 deletions
diff --git a/modules/hypernetworks/ui.py b/modules/hypernetworks/ui.py
index 2c6c0470..c2d4b51c 100644
--- a/modules/hypernetworks/ui.py
+++ b/modules/hypernetworks/ui.py
@@ -8,7 +8,8 @@ import modules.textual_inversion.textual_inversion
from modules import devices, sd_hijack, shared
from modules.hypernetworks import hypernetwork
-keys = list(hypernetwork.HypernetworkModule.activation_dict.keys())
+not_available = ["hardswish", "multiheadattention"]
+keys = list(x for x in hypernetwork.HypernetworkModule.activation_dict.keys() if x not in not_available)
def create_hypernetwork(name, enable_sizes, overwrite_old, layer_structure=None, activation_func=None, weight_init=None, add_layer_norm=False, use_dropout=False):
# Remove illegal characters from name.