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

Class GELU

models/transformer/wan/modules/t5.py:46–50  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

44
45
46class GELU(nn.Module):
47
48 def forward(self, x):
49 return 0.5 * x * (1.0 + torch.tanh(
50 math.sqrt(2.0 / math.pi) * (x + 0.044715 * torch.pow(x, 3.0))))
51
52
53class T5LayerNorm(nn.Module):

Callers 1

__init__Method · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected