#!/usr/bin/env python3
from diffusers import AutoPipelineForInpainting, AutoPipelineForText2Image, DiffusionPipeline, KandinskyPriorPipeline, DDPMScheduler, DDIMScheduler
import torch

cache_dir = DiffusionPipeline.download("kandinsky-community/kandinsky-2-2-decoder", torch_dtype=torch.float16, variant="fp16", load_connected_pipeline=True)
print(cache_dir)
pipe = DiffusionPipeline.from_pretrained(cache_dir, load_connected_pipeline=True)