| 132 | |
| 133 | |
| 134 | class LayerNormANE(LayerNormANE): |
| 135 | |
| 136 | def __init__(self, *args, **kwargs): |
| 137 | super().__init__(*args, **kwargs) |
| 138 | self._register_load_state_dict_pre_hook( |
| 139 | correct_for_bias_scale_order_inversion) |
| 140 | |
| 141 | |
| 142 | # Reference: https://github.com/huggingface/diffusers/blob/main/src/diffusers/models/unet_2d_condition.py |
nothing calls this directly
no outgoing calls
no test coverage detected