Update stable_cascade.py

stable_cascade.py

@@ -10,8 +10,8 @@ decoder = StableCascadeDecoderPipeline.from_pretrained("stabilityai/stable-casca
 prior.enable_model_cpu_offload()
 decoder.enable_model_cpu_offload()
 
-prior.prior = torch.compile(prior_pipeline.prior, mode="reduce-overhead", fullgraph=True)
-decoder.decoder = torch.compile(decoder_pipeline.decoder, mode="max-autotune", fullgraph=True)
+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")