Update README.md
Browse files
README.md
CHANGED
@@ -44,7 +44,7 @@ image = Image.open('<Your Conditioning Image Path>')
|
|
44 |
base_model = "runwayml/stable-diffusion-v1-5"
|
45 |
|
46 |
unet = UNet2DConditionModel.from_pretrained(
|
47 |
-
base_model, subfolder="unet"
|
48 |
)
|
49 |
controllora = ControlLoRAModel.from_pretrained(
|
50 |
"HighCWu/sd-controllora-face-landmarks", torch_dtype=torch.float16
|
|
|
44 |
base_model = "runwayml/stable-diffusion-v1-5"
|
45 |
|
46 |
unet = UNet2DConditionModel.from_pretrained(
|
47 |
+
base_model, subfolder="unet", torch_dtype=torch.float16
|
48 |
)
|
49 |
controllora = ControlLoRAModel.from_pretrained(
|
50 |
"HighCWu/sd-controllora-face-landmarks", torch_dtype=torch.float16
|