From cbb1ead60badd40487fa68ec82e725bc9d9ec804 Mon Sep 17 00:00:00 2001 From: Apoorva Kulkarni Date: Mon, 24 Jul 2023 11:07:53 -0700 Subject: [PATCH] docs: Add missing import statement in textual_inversion inference example (#4227) docs: Add missing import statement in textual_inversion inference instructions --- examples/textual_inversion/README.md | 1 + 1 file changed, 1 insertion(+) diff --git a/examples/textual_inversion/README.md b/examples/textual_inversion/README.md index fc231f8707..21bca526b5 100644 --- a/examples/textual_inversion/README.md +++ b/examples/textual_inversion/README.md @@ -100,6 +100,7 @@ Once you have trained a model using above command, the inference can be done sim ```python from diffusers import StableDiffusionPipeline +import torch model_id = "path-to-your-trained-model" pipe = StableDiffusionPipeline.from_pretrained(model_id,torch_dtype=torch.float16).to("cuda")