Update stable_cascade.py

stable_cascade.py

@@ -2,17 +2,10 @@ import torch, os
 from diffusers import StableCascadeDecoderPipeline, StableCascadePriorPipeline
 import gradio as gr
 
-torch._dynamo.config.suppress_errors = True
 os.environ['TOKENIZERS_PARALLELISM'] = 'false'
 
-prior = StableCascadePriorPipeline.from_pretrained("stabilityai/stable-cascade-prior", torch_dtype=torch.bfloat16)
-decoder = StableCascadeDecoderPipeline.from_pretrained("stabilityai/stable-cascade", torch_dtype=torch.bfloat16)
-
-prior.prior = torch.compile(prior.prior, mode="reduce-overhead", fullgraph=True)
-decoder.decoder = torch.compile(decoder.decoder, mode="max-autotune", fullgraph=True)
-
-prior.to("cuda")
-decoder.to("cuda")
+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")
 
 def generate_images(
     prompt="a photo of a girl",