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

Method forward

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

Source from the content-addressed store, hash-verified

172 return f"shape={tuple(self.scale.shape)}, eps={self.eps}"
173
174 def forward(self, x):
175 return rms_norm(x, self.scale, self.eps)
176
177
178class AdaRMSNorm(nn.Module):

Callers

nothing calls this directly

Calls 1

rms_normFunction · 0.85

Tested by

no test coverage detected