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
|
```python
|
||||||
|
import torch
|
||||||
from torch import autocast
|
from torch import autocast
|
||||||
import requests
|
import requests
|
||||||
from PIL import Image
|
from PIL import Image
|
||||||
|
@ -61,6 +62,7 @@ The `inpainting.py` script implements `StableDiffusionInpaintingPipeline`. This
|
||||||
### How to use it
|
### How to use it
|
||||||
|
|
||||||
```python
|
```python
|
||||||
|
import torch
|
||||||
from io import BytesIO
|
from io import BytesIO
|
||||||
|
|
||||||
from torch import autocast
|
from torch import autocast
|
||||||
|
|
Loading…
Reference in New Issue