From cdb81037b1ea3289753e947c1ae65c09a31bf327 Mon Sep 17 00:00:00 2001 From: Shyam Marjit <54628184+shyammarjit@users.noreply.github.com> Date: Thu, 25 Jan 2024 05:04:59 +0530 Subject: [PATCH] Updated README.md by changing DiffusionPipeline to StableDiffusionPipeline --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index 653613a..e3f854d 100644 --- a/README.md +++ b/README.md @@ -70,7 +70,7 @@ accelerate launch train_svdiff.py \ ### Inference ```python -from diffusers import DiffusionPipeline, DPMSolverMultistepScheduler +from diffusers import StableDiffusionPipeline, DPMSolverMultistepScheduler import torch from svdiff_pytorch import load_unet_for_svdiff, load_text_encoder_for_svdiff