| 259 | ).to(dtype=output_dtype) |
| 260 | |
| 261 | class LayerNorm(GGMLLayer, comfy.ops.manual_cast.LayerNorm): |
| 262 | def forward_ggml_cast_weights(self, input): |
| 263 | if self.weight is None: |
| 264 | return super().forward_comfy_cast_weights(input) |
| 265 | weight, bias = self.cast_bias_weight(input) |
| 266 | return torch.nn.functional.layer_norm(input, self.normalized_shape, weight, bias, self.eps) |
| 267 | |
| 268 | class GroupNorm(GGMLLayer, comfy.ops.manual_cast.GroupNorm): |
| 269 | def forward_ggml_cast_weights(self, input): |
nothing calls this directly
no outgoing calls
no test coverage detected