Method __init__

demo/Diffusion/models.py:195–204

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

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

    def get_model(self):
        return CLIPTextModel.from_pretrained(self.path,