patrickvonplaten commited on
Commit
90f150a
1 Parent(s): e8c88de

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -2
README.md CHANGED
@@ -39,8 +39,9 @@ This `stable-diffusion-2` model is resumed from [stable-diffusion-2-base](https:
39
  Using the [🤗's Diffusers library](https://github.com/huggingface/diffusers) to run Stable Diffusion 2 in a simple and efficient manner.
40
 
41
  ```bash
42
- pip install --upgrade git+https://github.com/huggingface/diffusers.git transformers accelerate scipy
43
  ```
 
44
  Running the pipeline (if you don't swap the scheduler it will run with the default DDIM, in this example we are swapping it to EulerDiscreteScheduler):
45
 
46
  ```python
@@ -54,7 +55,7 @@ pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler, to
54
  pipe = pipe.to("cuda")
55
 
56
  prompt = "a photo of an astronaut riding a horse on mars"
57
- image = pipe(prompt, height=768, width=768).images[0]
58
 
59
  image.save("astronaut_rides_horse.png")
60
  ```
39
  Using the [🤗's Diffusers library](https://github.com/huggingface/diffusers) to run Stable Diffusion 2 in a simple and efficient manner.
40
 
41
  ```bash
42
+ pip install diffusers transformers accelerate scipy safetensors
43
  ```
44
+
45
  Running the pipeline (if you don't swap the scheduler it will run with the default DDIM, in this example we are swapping it to EulerDiscreteScheduler):
46
 
47
  ```python
55
  pipe = pipe.to("cuda")
56
 
57
  prompt = "a photo of an astronaut riding a horse on mars"
58
+ image = pipe(prompt).images[0]
59
 
60
  image.save("astronaut_rides_horse.png")
61
  ```