Method __init__

demo/Diffusion/models.py:328–337

(self,
        hf_token,
        device,
        verbose,
        path,
        max_batch_size,
        embedding_dim
    )

class VAE(BaseModel):
    def __init__(self,
        hf_token,
        device,
        verbose,
        path,
        max_batch_size,
        embedding_dim
    ):
        super(VAE, self).__init__(hf_token, device=device, verbose=verbose, path=path, max_batch_size=max_batch_size, embedding_dim=embedding_dim)
        self.name = "VAE decoder"

    def get_model(self):
        vae = AutoencoderKL.from_pretrained(self.path,