Update stable_cascade.py

stable_cascade.py

@@ -5,7 +5,7 @@ import gradio as gr
 os.environ['TOKENIZERS_PARALLELISM'] = 'false'
 
 prior = StableCascadePriorPipeline.from_pretrained("stabilityai/stable-cascade-prior", torch_dtype=torch.bfloat16).to("cuda")
-decoder = StableCascadeDecoderPipeline.from_pretrained("stabilityai/stable-cascade", torch_dtype=torch.bfloat16).to("cuda")
+decoder = StableCascadeDecoderPipeline.from_pretrained("stabilityai/stable-cascade", torch_dtype=torch.float16).to("cuda")
 
 def generate_images(
     prompt="a photo of a girl",