Spaces:
Runtime error
Runtime error
Update stable_cascade.py
Browse files- stable_cascade.py +3 -3
stable_cascade.py
CHANGED
|
@@ -2,8 +2,8 @@ import torch
|
|
| 2 |
from diffusers import StableCascadeDecoderPipeline, StableCascadePriorPipeline
|
| 3 |
import gradio as gr
|
| 4 |
|
|
|
|
| 5 |
|
| 6 |
-
# Initialize the prior and decoder pipelines
|
| 7 |
prior = StableCascadePriorPipeline.from_pretrained("stabilityai/stable-cascade-prior", torch_dtype=torch.bfloat16)
|
| 8 |
decoder = StableCascadeDecoderPipeline.from_pretrained("stabilityai/stable-cascade", torch_dtype=torch.bfloat16)
|
| 9 |
|
|
@@ -90,7 +90,7 @@ def web_demo():
|
|
| 90 |
minimum=128,
|
| 91 |
maximum=1280,
|
| 92 |
step=32,
|
| 93 |
-
value=
|
| 94 |
label="Image Height",
|
| 95 |
)
|
| 96 |
|
|
@@ -98,7 +98,7 @@ def web_demo():
|
|
| 98 |
minimum=128,
|
| 99 |
maximum=1280,
|
| 100 |
step=32,
|
| 101 |
-
value=
|
| 102 |
label="Image Width",
|
| 103 |
)
|
| 104 |
with gr.Row():
|
|
|
|
| 2 |
from diffusers import StableCascadeDecoderPipeline, StableCascadePriorPipeline
|
| 3 |
import gradio as gr
|
| 4 |
|
| 5 |
+
torch._dynamo.config.suppress_errors = True
|
| 6 |
|
|
|
|
| 7 |
prior = StableCascadePriorPipeline.from_pretrained("stabilityai/stable-cascade-prior", torch_dtype=torch.bfloat16)
|
| 8 |
decoder = StableCascadeDecoderPipeline.from_pretrained("stabilityai/stable-cascade", torch_dtype=torch.bfloat16)
|
| 9 |
|
|
|
|
| 90 |
minimum=128,
|
| 91 |
maximum=1280,
|
| 92 |
step=32,
|
| 93 |
+
value=1024,
|
| 94 |
label="Image Height",
|
| 95 |
)
|
| 96 |
|
|
|
|
| 98 |
minimum=128,
|
| 99 |
maximum=1280,
|
| 100 |
step=32,
|
| 101 |
+
value=1024,
|
| 102 |
label="Image Width",
|
| 103 |
)
|
| 104 |
with gr.Row():
|