Views
No views yet




cd in the directory:1git clone https://github.com/HighCWu/control-lora-v3
2cd control-lora-v31# !pip install opencv-python transformers accelerate
2from diffusers import AutoencoderKL
3from diffusers.utils import load_image
4from model import UNet2DConditionModelEx
5from pipeline_sdxl import StableDiffusionXLControlLoraV3Pipeline
6import numpy as np
7import torch
8
9import cv2
10from PIL import Image
11
12prompt = "aerial view, a futuristic research complex in a bright foggy jungle, hard lighting"
13negative_prompt = "low quality, bad quality, sketches"
14
15# download an image
16image = load_image(
17 "https://hf.co/datasets/hf-internal-testing/diffusers-images/resolve/main/sd_controlnet/hf-logo.png"
18)
19
20# initialize the models and pipeline
21unet: UNet2DConditionModelEx = UNet2DConditionModelEx.from_pretrained(
22 "stabilityai/stable-diffusion-xl-base-1.0", subfolder="unet", torch_dtype=torch.float16
23)
24unet = unet.add_extra_conditions(["canny"])
25vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
26pipe = StableDiffusionXLControlLoraV3Pipeline.from_pretrained(
27 "stabilityai/stable-diffusion-xl-base-1.0", unet=unet, vae=vae, torch_dtype=torch.float16
28)
29# load attention processors
30pipe.load_lora_weights("HighCWu/sdxl-control-lora-v3-canny")
31pipe.enable_model_cpu_offload()
32
33# get canny image
34image = np.array(image)
35image = cv2.Canny(image, 100, 200)
36image = image[:, :, None]
37image = np.concatenate([image, image, image], axis=2)
38canny_image = Image.fromarray(image)
39
40# generate image
41image = pipe(
42 prompt, image=canny_image
43).images[0]
44image.show()