Function Normalize

ldm/modules/attention.py:88–89 · view source on GitHub ↗

(in_channels)

Source from the content-addressed store, hash-verified

86
87
88	def Normalize(in_channels):
89	return torch.nn.GroupNorm(num_groups=32, num_channels=in_channels, eps=1e-6, affine=True)
90
91
92	class SpatialSelfAttention(nn.Module):

__init__Method · 0.70

no outgoing calls

no test coverage detected