Update app.py

app.py

@@ -16,7 +16,7 @@ model_id = "Wan-AI/Wan2.1-T2V-1.3B-Diffusers"
 vae = AutoencoderKLWan.from_pretrained(model_id, subfolder="vae", torch_dtype=torch.float32)
 pipe = WanPipeline.from_pretrained(model_id, vae=vae, torch_dtype=torch.bfloat16)
 flow_shift = 5.0 #5.0  1.0 for image, 5.0 for 720P, 3.0 for 480P
-# pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config, flow_shift=flow_shift)
+pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config, flow_shift=flow_shift)
 
 # Configure DDIMScheduler with a beta schedule
 # pipe.scheduler = DDIMScheduler.from_config(
@@ -30,10 +30,10 @@ flow_shift = 5.0 #5.0  1.0 for image, 5.0 for 720P, 3.0 for 480P
 
 
 # Configure FlowMatchEulerDiscreteScheduler
-pipe.scheduler = FlowMatchEulerDiscreteScheduler.from_config(
-    pipe.scheduler.config,
-    flow_shift=flow_shift           # Retain flow_shift for WanPipeline compatibility
-)
+# pipe.scheduler = FlowMatchEulerDiscreteScheduler.from_config(
+#     pipe.scheduler.config,
+#     flow_shift=flow_shift           # Retain flow_shift for WanPipeline compatibility
+# )
 
 
 @spaces.GPU()