(self, x, t_emb=None)
| 182 | self.attns = nn.ModuleList(attns) |
| 183 | |
| 184 | def forward(self, x, t_emb=None): |
| 185 | x = self.nets[0](x, t_emb) |
| 186 | for attn, net in zip(self.attns, self.nets[1:]): |
| 187 | if attn: |
| 188 | x = attn(x) |
| 189 | x = net(x, t_emb) |
| 190 | return x |
| 191 | |
| 192 | class UpBlock_cond(nn.Module): |
| 193 | def __init__( |
nothing calls this directly
no outgoing calls
no test coverage detected