Merge pull request #3717 from benkyoujouzu/master

Add missing support for linear activation in hypernetwork
2026-02-02 06:20:04 +00:00 · 2022-10-29 07:30:14 +03:00
parent fc89495df3 b2a8b263b2
commit d3b4b9d7ec
1 changed files with 1 additions and 0 deletions
--- a/modules/hypernetworks/hypernetwork.py
+++ b/modules/hypernetworks/hypernetwork.py
@@ -25,6 +25,7 @@ from statistics import stdev, mean
 class HypernetworkModule(torch.nn.Module):
    multiplier = 1.0
    activation_dict = {
+        "linear": torch.nn.Identity,
        "relu": torch.nn.ReLU,
        "leakyrelu": torch.nn.LeakyReLU,
        "elu": torch.nn.ELU,