Update app.py

app.py

@@ -42,6 +42,7 @@ pipe = WanPipeline.from_pretrained(MODEL_ID,
     vae=vae,
     torch_dtype=torch.bfloat16,
 ).to('cuda')
+pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config, flow_shift=2.0)
 
 
 for i in range(3):