Unverified Commit d3b4b9d7 authored by AUTOMATIC1111's avatar AUTOMATIC1111 Committed by GitHub
Browse files

Merge pull request #3717 from benkyoujouzu/master 

Add missing support for linear activation in hypernetwork
parents fc89495d b2a8b263

modules/hypernetworks/hypernetwork.py

+1 −0
Original line number Diff line number Diff line
@@ -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,

CRA Git | Maintained and supported by SUSTech CRA and CCSE