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 |
+
"r3gm/controlnet-noobai-openpose-sdxl-fp16", torch_dtype=torch.float16
|
| 38 |
).to(device)
|
| 39 |
|
| 40 |
openpose_detector = OpenposeDetector.from_pretrained("lllyasviel/ControlNet").to(device)
|