From bc2ad5a66199e365c96b84969e22fe9739021da6 Mon Sep 17 00:00:00 2001
From: Patrick von Platen <patrick.v.platen@gmail.com>
Date: Wed, 31 Aug 2022 21:02:46 +0200
Subject: [PATCH] Improve README (#301)

---
 README.md | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/README.md b/README.md
index 4009ade519..4522c91f51 100644
--- a/README.md
+++ b/README.md
@@ -46,6 +46,7 @@ You need to accept the model license before downloading or using the Stable Diff
 ```python
 # make sure you're logged in with `huggingface-cli login`
 from torch import autocast
+import torch
 from diffusers import StableDiffusionPipeline, LMSDiscreteScheduler
 
 lms = LMSDiscreteScheduler(
@@ -77,6 +78,7 @@ The `StableDiffusionImg2ImgPipeline` lets you pass a text prompt and an initial
 ```python
 from torch import autocast
 import requests
+import torch
 from PIL import Image
 from io import BytesIO
 
@@ -116,6 +118,7 @@ The `StableDiffusionInpaintPipeline` lets you edit specific parts of an image by
 from io import BytesIO
 
 from torch import autocast
+import torch
 import requests
 import PIL