(self, h, force_not_quantize=False)
| 273 | return h |
| 274 | |
| 275 | def decode(self, h, force_not_quantize=False): |
| 276 | # also go through quantization layer |
| 277 | if not force_not_quantize: |
| 278 | quant, emb_loss, info = self.quantize(h) |
| 279 | else: |
| 280 | quant = h |
| 281 | quant = self.post_quant_conv(quant) |
| 282 | dec = self.decoder(quant) |
| 283 | return dec |
| 284 | |
| 285 | |
| 286 | class AutoencoderKL(pl.LightningModule): |