(self, x)
| 125 | self.conv_out = Conv2d(512, 8, 3, padding=1) |
| 126 | |
| 127 | def __call__(self, x): |
| 128 | x = self.conv_in(x) |
| 129 | |
| 130 | for i, l in enumerate(self.down): |
| 131 | for b in l['block']: x = b(x) |
| 132 | if 'downsample' in l: x = l['downsample']['conv'](x) |
| 133 | |
| 134 | x = self.mid(x) |
| 135 | return self.conv_out(self.norm_out(x).swish()) |
| 136 | |
| 137 | class AutoencoderKL: |
| 138 | def __init__(self): |
nothing calls this directly
no outgoing calls
no test coverage detected