Update app.py

app.py

@@ -1,5 +1,5 @@
 import torch
-from diffusers import UniPCMultistepScheduler, FlowMatchEulerDiscreteScheduler, DDIMScheduler
+from diffusers import UniPCMultistepScheduler, FlowMatchEulerDiscreteScheduler, DDIMScheduler, DPMSolverMultistepScheduler
 from diffusers import WanPipeline, AutoencoderKLWan  # Use Wan-specific VAE
 # from diffusers.hooks import apply_first_block_cache, FirstBlockCacheConfig
 from para_attn.first_block_cache.diffusers_adapters import apply_cache_on_pipe
@@ -15,7 +15,7 @@ import spaces
 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 = 1.0 #5.0  1.0 for image, 5.0 for 720P, 3.0 for 480P
+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)
 
 # Configure DDIMScheduler with a beta schedule