Update app_fast.py

app_fast.py

@@ -9,12 +9,13 @@ import numpy as np
 from PIL import Image
 import random
 
-MODEL_ID = "lopho/Wan2.2-T2V-A14B-Diffusers_nf4"
+
+MODEL_ID = "Wan-AI/Wan2.2-T2V-A14B-Diffusers"
 vae = AutoencoderKLWan.from_pretrained(MODEL_ID, subfolder="vae", torch_dtype=torch.float32)
 
 # Initialize pipelines
-text_to_video_pipe = WanPipeline.from_pretrained(MODEL_ID, vae=vae, torch_dtype=torch.float16)
-image_to_video_pipe = WanImageToVideoPipeline.from_pretrained(MODEL_ID, vae=vae, torch_dtype=torch.float16)
+text_to_video_pipe = WanPipeline.from_pretrained(MODEL_ID, vae=vae, torch_dtype=torch.bfloat16)
+image_to_video_pipe = WanImageToVideoPipeline.from_pretrained(MODEL_ID, vae=vae, torch_dtype=torch.bfloat16)
 
 for pipe in [text_to_video_pipe, image_to_video_pipe]:
     pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config, flow_shift=8.0)