Commit
•
2100baa
1
Parent(s):
81a84f4
Add import torch
Browse files
README.md
CHANGED
|
@@ -43,6 +43,7 @@ pip install diffusers transformers accelerate scipy safetensors
|
|
| 43 |
|
| 44 |
```python
|
| 45 |
from diffusers import StableDiffusionInpaintPipeline
|
|
|
|
| 46 |
pipe = StableDiffusionInpaintPipeline.from_pretrained(
|
| 47 |
"stabilityai/stable-diffusion-2-inpainting",
|
| 48 |
torch_dtype=torch.float16,
|
|
|
|
| 43 |
|
| 44 |
```python
|
| 45 |
from diffusers import StableDiffusionInpaintPipeline
|
| 46 |
+
import torch
|
| 47 |
pipe = StableDiffusionInpaintPipeline.from_pretrained(
|
| 48 |
"stabilityai/stable-diffusion-2-inpainting",
|
| 49 |
torch_dtype=torch.float16,
|