Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
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
259
class
WanSelfAttention(nn.Module):
Callers
nothing calls this directly
Calls
1
forward
Method · 0.45
Tested by
no test coverage detected