Update README.md

README.md

@@ -37,7 +37,6 @@ from PIL import Image
 from diffusers import StableDiffusionControlNetPipeline, UNet2DConditionModel, UniPCMultistepScheduler
 import torch
 from PIL import Image
-from diffusers.utils import load_image
 from models.controllora import ControlLoRAModel
 
 image = Image.open('<Your Conditioning Image Path>')
@@ -45,13 +44,13 @@ image = Image.open('<Your Conditioning Image Path>')
 unet = UNet2DConditionModel.from_pretrained(
     "runwayml/stable-diffusion-v1-5", subfolder="unet"
 )
-controllora = ControlLoRA.from_pretrained(
+controllora = ControlLoRAModel.from_pretrained(
     "HighCWu/HighCWu/sd-controllora-face-landmarks", torch_dtype=torch.float16
 )
 controllora.tie_weights(unet)
 
 pipe = StableDiffusionControlNetPipeline.from_pretrained(
-    "runwayml/stable-diffusion-v1-5", unet=unet, controlnet=controlnet, safety_checker=None, torch_dtype=torch.float16
+    "runwayml/stable-diffusion-v1-5", unet=unet, controlnet=controllora, safety_checker=None, torch_dtype=torch.float16
 )
 
 pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)