(self, h, force_not_quantize=False)
| 393 | return h |
| 394 | |
| 395 | def decode(self, h, force_not_quantize=False): |
| 396 | # also go through quantization layer |
| 397 | if not force_not_quantize: |
| 398 | quant, emb_loss, info = self.quantize(h) |
| 399 | else: |
| 400 | quant = h |
| 401 | quant = self.post_quant_conv(quant) |
| 402 | dec = self.decoder(quant) |
| 403 | return dec |
| 404 | |
| 405 | |
| 406 | class AutoencoderKL(pl.LightningModule): |