license: mit | |
tags: | |
- pytorch | |
- diffusers | |
- unconditional-image-generation | |
- diffusion-model-class | |
# 这个模型用于生成蝴蝶图像 | |
```python | |
from diffusers import DDPMPipeline | |
pipeline = DDPMPipeline.from_pretrained('DeepBird/my_custom_ddpm_train') | |
image = pipeline().images[0] | |
image | |
``` | |