import torch | |
from diffusers import StableDiffusionPipeline, LCMScheduler | |
model_id = "circulus/canvers-real-v3.8.1" | |
pipeline = StableDiffusionPipeline.from_pretrained(model_id, low_cpu_mem_usage=False) | |
pipeline.scheduler = LCMScheduler.from_config(pipeline.scheduler.config) | |
pipeline.load_lora_weights("latent-consistency/lcm-lora-sdv1-5") | |
pipeline.fuse_lora(lora_scale=1.0) | |
#pipeline = pipeline.to(dtype=torch.float16) | |
pipeline.save_pretrained(f"{model_id}-lcm") |