Update infer.py

infer.py

@@ -124,7 +124,7 @@ def parse_args():
 
 import torch
 from huggingface_hub import snapshot_download
-from diffusers import WanVideoPipeline
+from models.wan_video_pipeline.pipeline_wan_video import WanVideoPipeline
 from transformers import Wav2Vec2Processor, Wav2Vec2Model
 from models import FantasyTalkingAudioConditionModel  # adjust import if needed
 from model_manager import ModelManager  # assuming this exists in your repo