(self, inputs)
| 225 | |
| 226 | @torch.autocast('cuda', dtype=AUTOCAST_DTYPE) |
| 227 | def forward(self, inputs): |
| 228 | hidden_states, encoder_hidden_states, temb, height_width = inputs |
| 229 | |
| 230 | encoder_hidden_states, hidden_states = self.block( |
| 231 | hidden_states=hidden_states, |
| 232 | encoder_hidden_states=encoder_hidden_states, |
| 233 | temb=temb, |
| 234 | ) |
| 235 | |
| 236 | return make_contiguous(hidden_states, encoder_hidden_states, temb, height_width) |
| 237 | |
| 238 | |
| 239 | def concat_hidden_states(inputs): |
nothing calls this directly
no test coverage detected