Skip to content

Commit c19e73e

Browse files
kirill-fedyaninkashif
authored andcommitted
Fix the misaligned pipeline usage in dreamshaper docstrings (huggingface#5700)
Fix the misaligned pipeline usage
1 parent 2460754 commit c19e73e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/pipelines/latent_consistency_models/pipeline_latent_consistency_img2img.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,7 @@ def retrieve_latents(encoder_output, generator):
6060
>>> import torch
6161
>>> import PIL
6262
63-
>>> pipe = DiffusionPipeline.from_pretrained("SimianLuo/LCM_Dreamshaper_v7")
63+
>>> pipe = AutoPipelineForImage2Image.from_pretrained("SimianLuo/LCM_Dreamshaper_v7")
6464
>>> # To save GPU memory, torch.float16 can be used, but it may compromise image quality.
6565
>>> pipe.to(torch_device="cuda", torch_dtype=torch.float32)
6666

0 commit comments

Comments
 (0)