Adds missing torch imports to inpainting and image_to_image example (#265)

adds missing torch import to example
This commit is contained in:
Pulkit Mishra 2022-08-29 14:26:37 +05:30 committed by GitHub
parent 28f730520e
commit 16172c1c7e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions

View File

@ -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