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

Method forward

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

Source from the content-addressed store, hash-verified

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

Callers

nothing calls this directly

Calls 1

forwardMethod · 0.45

Tested by

no test coverage detected