MCPcopy Create free account
hub / github.com/MotrixLab/ViMoGen / forward

Method forward

models/transformer/wan/modules/tm2m_model.py:255–256  ·  view source on GitHub ↗
(self, x)

Source from the content-addressed store, hash-verified

253 super().__init__(dim, elementwise_affine=elementwise_affine, eps=eps)
254
255 def forward(self, x):
256 return super().forward(x.float()).type_as(x)
257
258
259class WanSelfAttention(nn.Module):

Callers

nothing calls this directly

Calls 1

forwardMethod · 0.45

Tested by

no test coverage detected