Update README.md
Browse files
README.md
CHANGED
|
@@ -50,7 +50,7 @@ This model can be used for entertainment purposes and as a generative art assist
|
|
| 50 |
```python
|
| 51 |
import torch
|
| 52 |
from torch import autocast
|
| 53 |
-
from diffusers import StableDiffusionPipeline
|
| 54 |
|
| 55 |
model_id = "hakurei/waifu-diffusion"
|
| 56 |
device = "cuda"
|
|
|
|
| 50 |
```python
|
| 51 |
import torch
|
| 52 |
from torch import autocast
|
| 53 |
+
from diffusers import StableDiffusionPipeline, DDIMScheduler
|
| 54 |
|
| 55 |
model_id = "hakurei/waifu-diffusion"
|
| 56 |
device = "cuda"
|