(self, quant)
| 73 | return quant, emb_loss, info |
| 74 | |
| 75 | def decode(self, quant): |
| 76 | quant = self.post_quant_conv(quant) |
| 77 | dec = self.decoder(quant) |
| 78 | return dec |
| 79 | |
| 80 | def decode_code(self, code_b, shape, channel_first=True): |
| 81 | quant_b = self.quantize.get_codebook_entry(code_b, shape, channel_first) |