Commit
•
90f150a
1
Parent(s):
e8c88de
Update README.md
Browse files
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
|
| 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
|
| 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 |
```
|