(self, x, time_emb)
| 125 | self.res_conv = nn.Conv2d(dim, dim_out, 1) if dim != dim_out else nn.Identity() |
| 126 | |
| 127 | def forward(self, x, time_emb): |
| 128 | h = self.block1(x) |
| 129 | h += self.mlp(time_emb)[:, :, None, None] |
| 130 | h = self.block2(h) |
| 131 | return h + self.res_conv(x) |
| 132 | |
| 133 | |
| 134 | class LinearAttention(nn.Module): |
nothing calls this directly
no outgoing calls
no test coverage detected