Skip to content

Commit 917a445

Browse files
PulkitMishranatolambert
authored andcommitted
Adds missing torch imports to inpainting and image_to_image example (#265)
adds missing torch import to example
1 parent 8c4c458 commit 917a445

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

examples/inference/readme.md

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@ The `image_to_image.py` script implements `StableDiffusionImg2ImgPipeline`. It l
1616

1717

1818
```python
19+
import torch
1920
from torch import autocast
2021
import requests
2122
from PIL import Image
@@ -61,6 +62,7 @@ The `inpainting.py` script implements `StableDiffusionInpaintingPipeline`. This
6162
### How to use it
6263

6364
```python
65+
import torch
6466
from io import BytesIO
6567

6668
from torch import autocast

0 commit comments

Comments
 (0)