Skip to content

Commit

Permalink
Update README.md (huggingface#5497)
Browse files Browse the repository at this point in the history
Right now, only "main" branch has this community pipeline code. So, adding it manually into pipeline
  • Loading branch information
anvilarth authored and kashif committed Nov 11, 2023
1 parent f3a27eb commit e43ddbe
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion examples/community/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down

0 comments on commit e43ddbe

Please sign in to comment.