(self)
| 136 | |
| 137 | class AutoencoderKL: |
| 138 | def __init__(self): |
| 139 | self.encoder = Encoder() |
| 140 | self.decoder = Decoder() |
| 141 | self.quant_conv = Conv2d(8, 8, 1) |
| 142 | self.post_quant_conv = Conv2d(4, 4, 1) |
| 143 | |
| 144 | def __call__(self, x): |
| 145 | latent = self.encoder(x) |