(self, hidden_states, temb=None)
| 380 | self.downsamplers = None |
| 381 | |
| 382 | def forward(self, hidden_states, temb=None): |
| 383 | output_states = () |
| 384 | |
| 385 | for resnet in self.resnets: |
| 386 | hidden_states = resnet(hidden_states, temb) |
| 387 | output_states += (hidden_states, ) |
| 388 | |
| 389 | if self.downsamplers is not None: |
| 390 | for downsampler in self.downsamplers: |
| 391 | hidden_states = downsampler(hidden_states) |
| 392 | |
| 393 | output_states = output_states + (hidden_states,) |
| 394 | |
| 395 | |
| 396 | return hidden_states, output_states |
| 397 | |
| 398 | |
| 399 | class ResnetBlock2D(nn.Module): |
nothing calls this directly
no outgoing calls
no test coverage detected