Modelpublic

diffusers/tools

sourceHugging Facecreativeml-openrail-mupdated 3y agoView on Hugging Face

11likes28downloads

run_kandinsky.py8 linesDownload Raw Back to root

1#!/usr/bin/env python32from diffusers import AutoPipelineForInpainting, AutoPipelineForText2Image, DiffusionPipeline, KandinskyPriorPipeline, DDPMScheduler, DDIMScheduler3import torch4 5cache_dir = DiffusionPipeline.download("kandinsky-community/kandinsky-2-2-decoder", torch_dtype=torch.float16, variant="fp16", load_connected_pipeline=True)6print(cache_dir)7pipe = DiffusionPipeline.from_pretrained(cache_dir, load_connected_pipeline=True)8