kadirnar commited on
Commit
edf4c40
1 Parent(s): 6fe7d03

Update stable_cascade.py

Browse files
Files changed (1) hide show
  1. 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=512,
94
  label="Image Height",
95
  )
96
 
@@ -98,7 +98,7 @@ def web_demo():
98
  minimum=128,
99
  maximum=1280,
100
  step=32,
101
- value=512,
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():