(self)
| 183 | tag_module(self.linear, "mapping") |
| 184 | |
| 185 | def extra_repr(self): |
| 186 | return f"eps={self.eps}," |
| 187 | |
| 188 | def forward(self, x, cond): |
| 189 | return rms_norm(x, self.linear(cond)[:, None, None, :] + 1, self.eps) |
nothing calls this directly
no outgoing calls
no test coverage detected