mirror of
https://github.com/huggingface/diffusers.git
synced 2026-01-27 17:22:53 +03:00
25 lines
641 B
Python
Executable File
25 lines
641 B
Python
Executable File
#!/usr/bin/env python3
|
|
import tempfile
|
|
|
|
from diffusers import GaussianDDPMScheduler, UNetModel
|
|
from modeling_ddpm import DDPM
|
|
|
|
|
|
unet = UNetModel.from_pretrained("fusing/ddpm_dummy")
|
|
sampler = GaussianDDPMScheduler.from_config("fusing/ddpm_dummy")
|
|
|
|
# compose Diffusion Pipeline
|
|
ddpm = DDPM(unet, sampler)
|
|
# generate / sample
|
|
image = ddpm()
|
|
print(image)
|
|
|
|
|
|
# save and load with 0 extra code (handled by general `DiffusionPipeline` class)
|
|
with tempfile.TemporaryDirectory() as tmpdirname:
|
|
ddpm.save_pretrained(tmpdirname)
|
|
print("Model saved")
|
|
ddpm_new = DDPM.from_pretrained(tmpdirname)
|
|
print("Model loaded")
|
|
print(ddpm_new)
|