#!/usr/bin/env python3
from diffusers.pipelines import StableDiffusionPipeline

from animatediff import get_dir

out_dir = get_dir("data/models/huggingface/stable-diffusion-v1-5")

pipeline = StableDiffusionPipeline.from_pretrained(
    "runwayml/stable-diffusion-v1-5",
    use_safetensors=True,
    kwargs=dict(safety_checker=None, requires_safety_checker=False),
)
pipeline.save_pretrained(
    save_directory=str(out_dir),
    safe_serialization=True,
)