Skip to content

GitLab

Unverified Commit acd87ae0 authored by Suraj Patil's avatar Suraj Patil Committed by GitHub
Browse files

update pipeline example

parent 28ba0ffa
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
README.md
View file @ acd87ae0
...@@ -84,12 +84,12 @@ image_pil.save("test.png") ...@@ -84,12 +84,12 @@ image_pil.save("test.png")
Example: Example:
```python ```python
from modeling_ddpm import DDPM from diffusers import DiffusionPipeline
import PIL.Image import PIL.Image
import numpy as np import numpy as np
# load model and scheduler # load model and scheduler
ddpm = DDPM.from_pretrained("fusing/ddpm-lsun-bedroom-pipe") ddpm = DiffusionPipeline.from_pretrained("fusing/ddpm-lsun-bedroom")
# run pipeline in inference (sample random noise and denoise) # run pipeline in inference (sample random noise and denoise)
image = ddpm() image = ddpm()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment