MCPcopy
hub / github.com/lllyasviel/Paints-UNDO / GroupNorm

Function GroupNorm

diffusers_vdm/vae.py:44–45  ·  view source on GitHub ↗
(in_channels, num_groups=32)

Source from the content-addressed store, hash-verified

42
43
44def GroupNorm(in_channels, num_groups=32):
45 return torch.nn.GroupNorm(num_groups=num_groups, num_channels=in_channels, eps=1e-6, affine=True)
46
47
48class DiagonalGaussianDistribution:

Callers 5

__init__Method · 0.85
__init__Method · 0.85
__init__Method · 0.85
__init__Method · 0.85
__init__Method · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected