File size: 458 Bytes
d0ffe9c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
#!/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,
)
|