diff options
author | AUTOMATIC1111 <16777216c@gmail.com> | 2022-11-05 13:15:00 +0000 |
---|---|---|
committer | GitHub <noreply@github.com> | 2022-11-05 13:15:00 +0000 |
commit | e96c4344952fba680aa0e7d919a8b3b1896fbeb5 (patch) | |
tree | f24476c5262f978d79c4bfff23deb02121d89b31 /modules/hypernetworks/ui.py | |
parent | 477c09f4e73dd21eb598dfe072e42b700fcb757d (diff) | |
parent | 7278897982bfb640ee95f144c97ed25fb3f77ea3 (diff) | |
download | stable-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.py | 2 |
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.
|