import torch
Browse files
README.md
CHANGED
|
@@ -44,6 +44,7 @@ Running the pipeline (if you don't swap the scheduler it will run with the defau
|
|
| 44 |
```python
|
| 45 |
from diffusers import DiffusionPipeline
|
| 46 |
from diffusers.utils import load_image
|
|
|
|
| 47 |
|
| 48 |
pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-2-1-unclip-small", torch_dtype=torch.float16)
|
| 49 |
pipe.to("cuda")
|
|
|
|
| 44 |
```python
|
| 45 |
from diffusers import DiffusionPipeline
|
| 46 |
from diffusers.utils import load_image
|
| 47 |
+
import torch
|
| 48 |
|
| 49 |
pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-2-1-unclip-small", torch_dtype=torch.float16)
|
| 50 |
pipe.to("cuda")
|