update

app.py

@@ -39,8 +39,8 @@ else:
 print(f"device={device}")
 #if torch.cuda.is_available():
 if device != "cpu":
-    prior_pipeline = StableCascadePriorPipeline.from_pretrained("stabilityai/stable-cascade-prior", torch_dtype=dtype)#.to(device)
-    decoder_pipeline = StableCascadeDecoderPipeline.from_pretrained("stabilityai/stable-cascade",  torch_dtype=dtype)#.to(device) 
+    prior_pipeline = StableCascadePriorPipeline.from_pretrained("stabilityai/stable-cascade-prior", torch_dtype=dtype, revision="refs/pr/2")#.to(device)
+    decoder_pipeline = StableCascadeDecoderPipeline.from_pretrained("stabilityai/stable-cascade",  torch_dtype=dtype, revision="refs/pr/44")#.to(device) 
 
     if ENABLE_CPU_OFFLOAD:
         prior_pipeline.enable_model_cpu_offload()

requirements.txt

@@ -1,6 +1,6 @@
 #git+https://github.com/kashif/diffusers.git@diffusers-yield-callback
-git+https://github.com/kashif/diffusers.git@a3dc21385b7386beb3dab3a9845962ede6765887
-#diffusers
+#git+https://github.com/kashif/diffusers.git@a3dc21385b7386beb3dab3a9845962ede6765887
+diffusers
 accelerate
 safetensors
 transformers