update

app.py

@@ -43,8 +43,12 @@ if device != "cpu":
     decoder_pipeline = StableCascadeDecoderPipeline.from_pretrained("stabilityai/stable-cascade",  torch_dtype=dtype)#.to(device) 
 
     if ENABLE_CPU_OFFLOAD:
-        prior_pipeline.enable_model_cpu_offload()
-        decoder_pipeline.enable_model_cpu_offload()
+        if device == "mps":
+            prior_pipeline.enable_attention_slicing()
+            decoder_pipeline.enable_attention_slicing()
+        else:
+            prior_pipeline.enable_model_cpu_offload()
+            decoder_pipeline.enable_model_cpu_offload()
     else:
         prior_pipeline.to(device)
         decoder_pipeline.to(device)
@@ -101,10 +105,10 @@ def generate(
         print("")
     
     #previewer.eval().requires_grad_(False).to(device).to(dtype)
-#    if device != "cpu":
-#        prior_pipeline.to(device)
-#        decoder_pipeline.to(device)
-#    
+    if device != "cpu":
+        prior_pipeline.to(device)
+        decoder_pipeline.to(device)
+
     generator = torch.Generator().manual_seed(seed)
     print("prior_num_inference_steps: ", prior_num_inference_steps)
     prior_output = prior_pipeline(