Method quantize

lvdm/models/autoencoder.py:215–218

(self, x, *args, **kwargs)

        return x

    def quantize(self, x, *args, **kwargs):
        if self.vq_interface:
            return x, None, [None, None, None]
        return x

    def forward(self, x, *args, **kwargs):
        return x