| #!/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) | |