Unverified Commit 11c12566 authored by Kirill's avatar Kirill Committed by GitHub
Browse files

Fix the misaligned pipeline usage in dreamshaper docstrings (#5700)

Fix the misaligned pipeline usage
parent 69996938
...@@ -60,7 +60,7 @@ EXAMPLE_DOC_STRING = """ ...@@ -60,7 +60,7 @@ EXAMPLE_DOC_STRING = """
>>> import torch >>> import torch
>>> import PIL >>> import PIL
>>> pipe = DiffusionPipeline.from_pretrained("SimianLuo/LCM_Dreamshaper_v7") >>> pipe = AutoPipelineForImage2Image.from_pretrained("SimianLuo/LCM_Dreamshaper_v7")
>>> # To save GPU memory, torch.float16 can be used, but it may compromise image quality. >>> # To save GPU memory, torch.float16 can be used, but it may compromise image quality.
>>> pipe.to(torch_device="cuda", torch_dtype=torch.float32) >>> pipe.to(torch_device="cuda", torch_dtype=torch.float32)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment