File size: 441 Bytes
a032108 6c27fdd a032108 6c27fdd |
1 2 3 4 5 6 7 8 |
#!/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)
|