Update app.py
Browse files
app.py
CHANGED
@@ -34,7 +34,7 @@ pipe = StableDiffusionXLPipeline.from_pretrained(
|
|
34 |
pipe.to(device)
|
35 |
|
36 |
controlnet_openpose = ControlNetModel.from_pretrained(
|
37 |
-
"
|
38 |
).to(device)
|
39 |
|
40 |
openpose_detector = OpenposeDetector.from_pretrained("lllyasviel/ControlNet").to(device)
|
|
|
34 |
pipe.to(device)
|
35 |
|
36 |
controlnet_openpose = ControlNetModel.from_pretrained(
|
37 |
+
"lllyasviel/control_v11p_sd15_openpose", torch_dtype=torch.float16
|
38 |
).to(device)
|
39 |
|
40 |
openpose_detector = OpenposeDetector.from_pretrained("lllyasviel/ControlNet").to(device)
|