Adds missing torch imports to inpainting and image_to_image example (#265)
adds missing torch import to example
This commit is contained in:
parent
28f730520e
commit
16172c1c7e
|
@ -16,6 +16,7 @@ The `image_to_image.py` script implements `StableDiffusionImg2ImgPipeline`. It l
|
|||
|
||||
|
||||
```python
|
||||
import torch
|
||||
from torch import autocast
|
||||
import requests
|
||||
from PIL import Image
|
||||
|
@ -61,6 +62,7 @@ The `inpainting.py` script implements `StableDiffusionInpaintingPipeline`. This
|
|||
### How to use it
|
||||
|
||||
```python
|
||||
import torch
|
||||
from io import BytesIO
|
||||
|
||||
from torch import autocast
|
||||
|
|
Loading…
Reference in New Issue