Method forward

k_diffusion/models/modules.py:188–189

(self, x, cond)

        return f"eps={self.eps},"

    def forward(self, x, cond):
        return rms_norm(x, self.linear(cond)[:, None, None, :] + 1, self.eps)
    

# Rotary position embeddings