samedii / perceptor

Modular image generation library
Other
5 stars 1 forks source link

Add diffusers face model #34

Open samedii opened 2 years ago

samedii commented 2 years ago

https://github.com/huggingface/diffusers

from diffusers import DDPMPipeline, DDIMPipeline, PNDMPipeline

model_id = "google/ddpm-celebahq-256"

# load model and scheduler
ddpm = DDPMPipeline.from_pretrained(model_id) # you can replace DDPMPipeline with DDIMPipeline or PNDMPipeline for faster inference