File size: 717 Bytes
c945057 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 |
#!/usr/bin/env python
import os
import sys
import torch
from diffusers import StableDiffusionPipeline
from diffusers import (
AutoencoderKL,
DDIMScheduler,
DDPMScheduler,
StableDiffusionPipeline,
UNet2DConditionModel,
)
from transformers import CLIPTextModel, CLIPTokenizer
cache_dir = "ssd-cache"
vae_cache_dir = "sd-vae-ft-mse-cache"
os.makedirs(cache_dir, exist_ok=True)
os.makedirs(vae_cache_dir, exist_ok=True)
pipe = StableDiffusionPipeline.from_pretrained(
"segmind/SSD-1B-fp32",
)
pipe.save_pretrained(cache_dir)
pretrained_vae = AutoencoderKL.from_pretrained(
"madebyollin/sdxl-vae-fp16-fix", subfolder=None, revision=None
)
pretrained_vae.save_pretrained(vae_cache_dir)
|