Skip to content

GitLab

You need to sign in or sign up before continuing.
Unverified Commit acd87ae0 authored by Suraj Patil's avatar Suraj Patil Committed by GitHub
Browse files

update pipeline example

parent 28ba0ffa
Show 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