Update README.md

README.md

@@ -41,8 +41,10 @@ from models.controllora import ControlLoRAModel
 
 image = Image.open('<Your Conditioning Image Path>')
 
+base_model = "runwayml/stable-diffusion-v1-5"
+
 unet = UNet2DConditionModel.from_pretrained(
-    "runwayml/stable-diffusion-v1-5", subfolder="unet"
+    base_model, subfolder="unet"
 )
 controllora = ControlLoRAModel.from_pretrained(
     "HighCWu/HighCWu/sd-controllora-face-landmarks", torch_dtype=torch.float16
@@ -50,7 +52,7 @@ controllora = ControlLoRAModel.from_pretrained(
 controllora.tie_weights(unet)
 
 pipe = StableDiffusionControlNetPipeline.from_pretrained(
-    "runwayml/stable-diffusion-v1-5", unet=unet, controlnet=controllora, safety_checker=None, torch_dtype=torch.float16
+    base_model, unet=unet, controlnet=controllora, safety_checker=None, torch_dtype=torch.float16
 )
 
 pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)