Function make_CLIP

demo/Diffusion/models.py:257–259

(version, hf_token, device, verbose, max_batch_size, inpaint=False)

        return opt_onnx_graph

def make_CLIP(version, hf_token, device, verbose, max_batch_size, inpaint=False):
    return CLIP(hf_token=hf_token, device=device, verbose=verbose, path=get_path(version, inpaint=inpaint),
                max_batch_size=max_batch_size, embedding_dim=get_embedding_dim(version))

class UNet(BaseModel):
    def __init__(self,