Update app.py

app.py

@@ -40,8 +40,9 @@ LORA_FILENAME = "Wan21_CausVid_14B_T2V_lora_rank32.safetensors"
 
 vae = AutoencoderKLWan.from_pretrained(MODEL_ID, subfolder="vae", torch_dtype=torch.float32)
 wan_path = hf_hub_download(repo_id=SUB_MODEL_ID, filename=SUB_MODEL_FILENAME)
-pipe = NAGWanPipeline.from_single_file(
-    wan_path, vae=vae, torch_dtype=torch.bfloat16
+transformer = AutoModel.from_pretrained(wan_path, torch_dtype=torch.bfloat16, use_safetensors=True)
+pipe = NAGWanPipeline.from_pretrained(
+    MODEL_ID, vae=vae, transformer=transformer, torch_dtype=torch.bfloat16
 )
 pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config, flow_shift=5.0)
 pipe.to("cuda")