Update README.md

README.md

@@ -8,7 +8,7 @@ tags:
 - image-to-image
 - diffusers
 - controlnet
-- controllora
+- control-lora
 ---
     
 # ControlLoRA - Face Landmarks Version
@@ -37,7 +37,7 @@ from PIL import Image
 from diffusers import StableDiffusionControlNetPipeline, UNet2DConditionModel, UniPCMultistepScheduler
 import torch
 from PIL import Image
-from models.controllora import ControlLoRAModel
+from models.control_lora import ControlLoRAModel
 
 image = Image.open('<Your Conditioning Image Path>')
 
@@ -46,13 +46,13 @@ base_model = "runwayml/stable-diffusion-v1-5"
 unet = UNet2DConditionModel.from_pretrained(
     base_model, subfolder="unet", torch_dtype=torch.float16
 )
-controllora = ControlLoRAModel.from_pretrained(
-    "HighCWu/sd-controllora-face-landmarks", torch_dtype=torch.float16
+control_lora = ControlLoRAModel.from_pretrained(
+    "HighCWu/sd-control-lora-face-landmarks", torch_dtype=torch.float16
 )
-controllora.tie_weights(unet)
+control_lora.tie_weights(unet)
 
 pipe = StableDiffusionControlNetPipeline.from_pretrained(
-    base_model, unet=unet, controlnet=controllora, safety_checker=None, torch_dtype=torch.float16
+    base_model, unet=unet, controlnet=control_lora, safety_checker=None, torch_dtype=torch.float16
 )
 
 pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)