Modelpublic

diffusers/tools

sourceHugging Facecreativeml-openrail-mupdated 3y agoView on Hugging Face

11likes28downloads

load_lora.py13 linesDownload Raw Back to root

1#!/usr/bin/env python32import torch3from diffusers import StableDiffusionPipeline, DPMSolverMultistepScheduler4 5pipeline = StableDiffusionPipeline.from_pretrained(6    "runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16, safety_checker=None7).to("cuda")8pipeline.scheduler = DPMSolverMultistepScheduler.from_config(9    pipeline.scheduler.config, use_karras_sigmas=True10)11 12pipeline.load_lora_weights("sayakpaul/sd-model-finetuned-lora-t4")13