Skip to content

GitLab

Unverified Commit 16172c1c authored by Pulkit Mishra's avatar Pulkit Mishra Committed by GitHub
Browse files

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

adds missing torch import to example
parent 28f73052
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
examples/inference/readme.md
View file @ 16172c1c
...@@ -16,6 +16,7 @@ The `image_to_image.py` script implements `StableDiffusionImg2ImgPipeline`. It l ...@@ -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 ...@@ -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
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment