(self)
| 1027 | torch.nn.init.zeros_(self.adaln_modulation_mlp[1].weight) |
| 1028 | |
| 1029 | def init_weights(self) -> None: |
| 1030 | self.reset_parameters() |
| 1031 | self.self_attn.init_weights() |
| 1032 | self.cross_attn.init_weights() |
| 1033 | self.mlp.init_weights() |
| 1034 | |
| 1035 | def forward( |
| 1036 | self, |
no test coverage detected