Method forward

k_diffusion/models/modules.py:160–162 · view source on GitHub ↗

(self, x)

Source from the content-addressed store, hash-verified

158	self.out_features = out_features
159
160	def forward(self, x):
161	flops.op(flops.op_linear, x.shape, self.weight.shape)
162	return linear_geglu(x, self.weight, self.bias)
163
164
165	class RMSNorm(nn.Module):

nothing calls this directly

linear_gegluFunction · 0.85

opMethod · 0.80

no test coverage detected