Skip to content

GitLab

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
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/community/README.md
View file @ 16851efa
...@@ -2165,7 +2165,7 @@ The model can be used with `diffusers` as follows: ...@@ -2165,7 +2165,7 @@ The model can be used with `diffusers` as follows:
from diffusers import DiffusionPipeline from diffusers import DiffusionPipeline
import torch 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. # 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