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 |
)
|