Skip to content

GitLab

Commit 1a310200 authored by comfyanonymous's avatar comfyanonymous
Browse files

Support softsign hypernetwork.

parent bae4fb4a
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
comfy_extras/nodes_hypernetwork.py
View file @ 1a310200
...@@ -18,6 +18,7 @@ def load_hypernetwork_patch(path, strength): ...@@ -18,6 +18,7 @@ def load_hypernetwork_patch(path, strength):
"swish": torch.nn.Hardswish, "swish": torch.nn.Hardswish,
"tanh": torch.nn.Tanh, "tanh": torch.nn.Tanh,
"sigmoid": torch.nn.Sigmoid, "sigmoid": torch.nn.Sigmoid,
"softsign": torch.nn.Softsign,
} }
if activation_func not in valid_activation: if activation_func not in valid_activation:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment