(self, x, noise_cond)
| 647 | |
| 648 | |
| 649 | def forward(self, x, noise_cond): |
| 650 | x = rearrange(x, 'b c h w -> b h w c') |
| 651 | x=self.proj_in(x) |
| 652 | x=self.ff(x, noise_cond) |
| 653 | x = rearrange(x, 'b h w c -> b c h w') |
| 654 | |
| 655 | return x |
nothing calls this directly
no outgoing calls
no test coverage detected