code: def __init__(self, embed_dim, *args, **kwargs):
      super().__init__(embed_dim=embed_dim, *args, **kwargs)
      self.embed_dim = embed_dim
  
imports: import torch
  import pytorch_lightning
  import torch.nn.functional
  from contextlib import contextmanager
  from taming.modules.vqvae.quantize import VectorQuantizer2
  from ldm.modules.diffusionmodules.model import Encoder, Decoder
  from ldm.modules.distributions.distributions import DiagonalGaussianDistribution
  from ldm.util import instantiate_from_config
label_desc: Use image node_id 1 to create a new VQModelInterface object from inherited base classes: VQModel with example: obj = VQModelInterface(embed_dim)