Method __init__

code/dc_ldm/models/autoencoder.py:386–388

(self, embed_dim, *args, **kwargs)

class VQModelInterface(VQModel):
    def __init__(self, embed_dim, *args, **kwargs):
        super().__init__(embed_dim=embed_dim, *args, **kwargs)
        self.embed_dim = embed_dim

    def encode(self, x):
        h = self.encoder(x)