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
Hide 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
```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
......
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