rippertnt's picture
Upload 4 files
2d5a280 verified
raw
history blame contribute delete
No virus
464 Bytes
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")