Support softsign hypernetwork.

2023-05-05 00:16:57 -04:00 · 2023-05-05 00:16:57 -04:00 · 1a31020081
parent bae4fb4a9d
commit 1a31020081
1 changed files with 1 additions and 0 deletions
--- a/comfy_extras/nodes_hypernetwork.py
+++ b/comfy_extras/nodes_hypernetwork.py
@ -18,6 +18,7 @@ def load_hypernetwork_patch(path, strength):
        "swish": torch.nn.Hardswish,
        "tanh": torch.nn.Tanh,
        "sigmoid": torch.nn.Sigmoid,
+        "softsign": torch.nn.Softsign,
    }

    if activation_func not in valid_activation: