back to real name StableCascadeDecoderPipeline

handler.py

@@ -4,7 +4,7 @@ from PIL import Image
 from io import BytesIO
 from diffusers.pipelines.stable_diffusion import StableDiffusionSafetyChecker
 from diffusers import StableDiffusionPipeline
-from diffusers import StableCascadePipeline, StableCascadePriorPipeline
+from diffusers import StableCascadeDecoderPipeline, StableCascadePriorPipeline
 
 import torch
 
@@ -21,7 +21,7 @@ class EndpointHandler():
          self.stable_diffusion_id = "Lykon/dreamshaper-8"
 
          self.prior_pipeline = StableCascadePriorPipeline.from_pretrained("stabilityai/stable-cascade-prior", torch_dtype=dtype)#.to(device)
-         self.decoder_pipeline = StableCascadePipeline.from_pretrained("stabilityai/stable-cascade",  torch_dtype=dtype)#.to(device) 
+         self.decoder_pipeline = StableCascadeDecoderPipeline.from_pretrained("stabilityai/stable-cascade",  torch_dtype=dtype)#.to(device) 
 
 
          self.generator = torch.Generator(device=device.type).manual_seed(3)