mirrors/stable-diffusion-webui
1
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2026-01-06 01:02:02 -08:00
Code Issues Projects Releases Packages Wiki Activity

Merge pull request #3717 from benkyoujouzu/master

Browse source 
Add missing support for linear activation in hypernetwork
This commit is contained in:
AUTOMATIC1111 2022-10-29 07:30:14 +03:00 • committed by GitHub
commit d3b4b9d7ec
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
modules/hypernetworks/hypernetwork.py
View file

@ -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,