Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -62,7 +62,7 @@ pipe = StableDiffusion3Pipeline.from_pretrained("ford442/stable-diffusion-3.5-me
|
|
| 62 |
|
| 63 |
#pipe.scheduler.config.requires_aesthetics_score = False
|
| 64 |
#pipe.enable_model_cpu_offload()
|
| 65 |
-
pipe.to(
|
| 66 |
#pipe = torch.compile(pipe)
|
| 67 |
# pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config, beta_schedule="scaled_linear")
|
| 68 |
|
|
@@ -264,7 +264,7 @@ with gr.Blocks(css=css) as demo:
|
|
| 264 |
minimum=0.0,
|
| 265 |
maximum=10.0,
|
| 266 |
step=0.1,
|
| 267 |
-
value=
|
| 268 |
)
|
| 269 |
num_inference_steps = gr.Slider(
|
| 270 |
label="Number of inference steps",
|
|
|
|
| 62 |
|
| 63 |
#pipe.scheduler.config.requires_aesthetics_score = False
|
| 64 |
#pipe.enable_model_cpu_offload()
|
| 65 |
+
pipe.to(torch_device="cuda", torch_dtype=torch.bfloat16)
|
| 66 |
#pipe = torch.compile(pipe)
|
| 67 |
# pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config, beta_schedule="scaled_linear")
|
| 68 |
|
|
|
|
| 264 |
minimum=0.0,
|
| 265 |
maximum=10.0,
|
| 266 |
step=0.1,
|
| 267 |
+
value=4.2, # Replace with defaults that work for your model
|
| 268 |
)
|
| 269 |
num_inference_steps = gr.Slider(
|
| 270 |
label="Number of inference steps",
|