Hybrid SD: Edge-Cloud Collaborative Inference for Stable Diffusion Models
Paper • 2408.06646 • Published • 1
How to use cqyan/hybrid-sd-v1-4-lcm with Diffusers:
pip install -U diffusers transformers accelerate
import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("cqyan/hybrid-sd-v1-4-lcm", dtype=torch.bfloat16, device_map="cuda")
prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
image = pipe(prompt).images[0]Hybrid-SD-v1-4-LCM is a latent consistency model based on stable-diffusion-v1-4.
@article{yan2024hybrid,
title={Hybrid SD: Edge-Cloud Collaborative Inference for Stable Diffusion Models},
author={Yan, Chenqian and Liu, Songwei and Liu, Hongjian and Peng, Xurui and Wang, Xiaojian and Chen, Fangming and Fu, Lean and Mei, Xing},
journal={arXiv preprint arXiv:2408.06646},
year={2024}
}