Update README.md

README.md

@@ -151,7 +151,8 @@ pipe = StableDiffusionControlLoraV3Pipeline.from_pretrained(
     "runwayml/stable-diffusion-v1-5", unet=unet, torch_dtype=torch.float16
 )
 # load attention processors
-pipe.load_lora_weights("HighCWu/sd-control-lora-v3-canny")
+# pipe.load_lora_weights("HighCWu/sd-control-lora-v3-canny")
+pipe.load_lora_weights("HighCWu/control-lora-v3", subfolder="sd-control-lora-v3-canny-half_skip_attn-rank16-conv_in-rank64")
 
 # speed up diffusion process with faster scheduler and memory optimization
 pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
@@ -200,7 +201,8 @@ pipe = StableDiffusionXLControlLoraV3Pipeline.from_pretrained(
     "stabilityai/stable-diffusion-xl-base-1.0", unet=unet, vae=vae, torch_dtype=torch.float16
 )
 # load attention processors
-pipe.load_lora_weights("HighCWu/sdxl-control-lora-v3-canny")
+# pipe.load_lora_weights("HighCWu/sdxl-control-lora-v3-canny")
+pipe.load_lora_weights("HighCWu/control-lora-v3", subfolder="sdxl-control-lora-v3-canny-half_skip_attn-rank16-conv_in-rank64")
 pipe.enable_model_cpu_offload()
 
 # get canny image