Unverified Commit 16851efa authored by Andrei Filatov's avatar Andrei Filatov Committed by GitHub
Browse files

Update README.md (#5497)

Right now, only "main" branch has this community pipeline code. So, adding it manually into pipeline
parent 0eac9cd0
......@@ -2165,7 +2165,7 @@ The model can be used with `diffusers` as follows:
from diffusers import DiffusionPipeline
import torch
pipe = DiffusionPipeline.from_pretrained("SimianLuo/LCM_Dreamshaper_v7", custom_pipeline="latent_consistency_txt2img")
pipe = DiffusionPipeline.from_pretrained("SimianLuo/LCM_Dreamshaper_v7", custom_pipeline="latent_consistency_txt2img", custom_revision="main")
# To save GPU memory, torch.float16 can be used, but it may compromise image quality.
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