Views
No views yet
$ git clone https://github.com/HighCWu/control-lora-v2$ cd control-lora-v21from PIL import Image
2from diffusers import StableDiffusionControlNetPipeline, UNet2DConditionModel, UniPCMultistepScheduler
3import torch
4from PIL import Image
5from models.controllora import ControlLoRAModel
6
7image = Image.open('<Your Conditioning Image Path>')
8
9base_model = "runwayml/stable-diffusion-v1-5"
10
11unet = UNet2DConditionModel.from_pretrained(
12 base_model, subfolder="unet", torch_dtype=torch.float16
13)
14controllora: ControlLoRAModel = ControlLoRAModel.from_pretrained(
15 "shiviguptta/sd-controllora-face-landmarks", torch_dtype=torch.float16
16)
17controllora.tie_weights(unet)
18
19pipe = StableDiffusionControlNetPipeline.from_pretrained(
20 base_model, unet=unet, controlnet=controllora, safety_checker=None, torch_dtype=torch.float16
21)
22
23pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
24
25# Remove if you do not have xformers installed
26# see https://huggingface.co/docs/diffusers/v0.13.0/en/optimization/xformers#installing-xformers
27# for installation instructions
28pipe.enable_xformers_memory_efficient_attention()
29
30pipe.enable_model_cpu_offload()
31
32image = pipe("Girl smiling, professional dslr photograph, high quality", image, num_inference_steps=20).images[0]
33
34image.show()

