aboutsummaryrefslogtreecommitdiffstats
path: root/modules/hypernetworks/ui.py
diff options
context:
space:
mode:
authorAUTOMATIC1111 <16777216c@gmail.com>2022-11-05 13:15:00 +0000
committerGitHub <noreply@github.com>2022-11-05 13:15:00 +0000
commite96c4344952fba680aa0e7d919a8b3b1896fbeb5 (patch)
treef24476c5262f978d79c4bfff23deb02121d89b31 /modules/hypernetworks/ui.py
parent477c09f4e73dd21eb598dfe072e42b700fcb757d (diff)
parent7278897982bfb640ee95f144c97ed25fb3f77ea3 (diff)
downloadstable-diffusion-webui-gfx803-e96c4344952fba680aa0e7d919a8b3b1896fbeb5.tar.gz
stable-diffusion-webui-gfx803-e96c4344952fba680aa0e7d919a8b3b1896fbeb5.tar.bz2
stable-diffusion-webui-gfx803-e96c4344952fba680aa0e7d919a8b3b1896fbeb5.zip
Merge pull request #3975 from aria1th/force-push-patch-13
Save/loading AdamW optimizer (for hypernetworks)
Diffstat (limited to 'modules/hypernetworks/ui.py')
-rw-r--r--modules/hypernetworks/ui.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/modules/hypernetworks/ui.py b/modules/hypernetworks/ui.py
index aad09ffc..c2d4b51c 100644
--- a/modules/hypernetworks/ui.py
+++ b/modules/hypernetworks/ui.py
@@ -9,7 +9,7 @@ from modules import devices, sd_hijack, shared
from modules.hypernetworks import hypernetwork
not_available = ["hardswish", "multiheadattention"]
-keys = ["linear"] + list(x for x in hypernetwork.HypernetworkModule.activation_dict.keys() if x not in not_available)
+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.