Spaces:
Running
Running
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():
|