MCPcopy Create free account
hub / github.com/Meshcapade/difflocks / forward

Method forward

k_diffusion/models/modules.py:150–152  ·  view source on GitHub ↗
(self, x)

Source from the content-addressed store, hash-verified

148
149class Linear(nn.Linear):
150 def forward(self, x):
151 flops.op(flops.op_linear, x.shape, self.weight.shape)
152 return super().forward(x)
153
154
155class LinearGEGLU(nn.Linear):

Callers

nothing calls this directly

Calls 2

opMethod · 0.80
forwardMethod · 0.45

Tested by

no test coverage detected