Function nonlinearity

ldm/modules/diffusionmodules/model.py:33–35 · view source on GitHub ↗

(x)

Source from the content-addressed store, hash-verified

31
32
33	def nonlinearity(x):
34	# swish
35	return x*torch.sigmoid(x)
36
37
38	def Normalize(in_channels, num_groups=32):

forwardMethod · 0.85

no outgoing calls

no test coverage detected