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)
|
@@ -273,7 +273,7 @@ with gr.Blocks(css=css) as demo:
|
|
273 |
num_inference_steps = gr.Slider(
|
274 |
label="Number of inference steps",
|
275 |
minimum=1,
|
276 |
-
maximum=
|
277 |
step=1,
|
278 |
value=60,
|
279 |
)
|
|
|
34 |
pipe.to(device)
|
35 |
|
36 |
controlnet_openpose = ControlNetModel.from_pretrained(
|
37 |
+
"thibaud/controlnet-openpose-sdxl-1.0", torch_dtype=torch.float16
|
38 |
).to(device)
|
39 |
|
40 |
openpose_detector = OpenposeDetector.from_pretrained("lllyasviel/ControlNet").to(device)
|
|
|
273 |
num_inference_steps = gr.Slider(
|
274 |
label="Number of inference steps",
|
275 |
minimum=1,
|
276 |
+
maximum=1000, # More typical steps for SDXL (20-50 usually sufficient)
|
277 |
step=1,
|
278 |
value=60,
|
279 |
)
|