Class LayerNorm

wan/models/wan_image_encoder.py:49–52 · view source on GitHub ↗

Source from the content-addressed store, hash-verified

47
48
49	class LayerNorm(nn.LayerNorm):
50
51	def forward(self, x):
52	return super().forward(x.float()).type_as(x)
53
54
55	class SelfAttention(nn.Module):

__init__Method · 0.85

no outgoing calls

no test coverage detected