From 16172c1c7ef6ec721bfe4d0787313519157749a1 Mon Sep 17 00:00:00 2001 From: Pulkit Mishra Date: Mon, 29 Aug 2022 14:26:37 +0530 Subject: [PATCH] Adds missing torch imports to inpainting and image_to_image example (#265) adds missing torch import to example --- examples/inference/readme.md | 2 ++ 1 file changed, 2 insertions(+) diff --git a/examples/inference/readme.md b/examples/inference/readme.md index affe9554cb..37e9d859a3 100644 --- a/examples/inference/readme.md +++ b/examples/inference/readme.md @@ -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