Update app.py

app.py

@@ -5,6 +5,7 @@ os.system('pip install --upgrade --pre --extra-index-url https://download.pytorc
 # --- 1. Model Download and Setup (Diffusers Backend) ---
 import spaces
 import torch
+from diffusers import FlowMatchEulerDiscreteScheduler
 from diffusers.pipelines.wan.pipeline_wan_i2v import WanImageToVideoPipeline
 from diffusers.models.transformers.transformer_wan import WanTransformer3DModel
 from diffusers.utils.export_utils import export_to_video
@@ -53,7 +54,11 @@ pipe = WanImageToVideoPipeline.from_pretrained(
         device_map='cuda',
     ),
     torch_dtype=torch.bfloat16,
-).to('cuda')
+)
+pipe.scheduler = FlowMatchEulerDiscreteScheduler.from_config(pipe.scheduler.config, shift=8.0)
+pipe.to('cuda')
+
+
 
 print("Optimizing pipeline...")
 for i in range(3):