Update OmniAvatar/wan_video.py

OmniAvatar/wan_video.py

@@ -4,7 +4,6 @@ from .models.wan_video_dit import WanModel
 from .models.wan_video_text_encoder import WanTextEncoder
 from .models.wan_video_vae import WanVideoVAE
 from .schedulers.flow_match import FlowMatchScheduler
-
 from .base import BasePipeline
 from .prompters import WanPrompter
 import torch, os
@@ -18,16 +17,12 @@ from .models.wan_video_text_encoder import T5RelativeEmbedding, T5LayerNorm
 from .models.wan_video_dit import RMSNorm
 from .models.wan_video_vae import RMS_norm, CausalConv3d, Upsample
 
-from diffusers import UniPCMultistepScheduler
 
 class WanVideoPipeline(BasePipeline):
 
     def __init__(self, device="cuda", torch_dtype=torch.float16, tokenizer_path=None):
         super().__init__(device=device, torch_dtype=torch_dtype)
         self.scheduler = FlowMatchScheduler(shift=5, sigma_min=0.0, extra_one_step=True)
-
-        self.scheduler = UniPCMultistepScheduler.from_config(self.scheduler.config, flow_shift=8.0)
-        
         self.prompter = WanPrompter(tokenizer_path=tokenizer_path)
         self.text_encoder: WanTextEncoder = None
         self.image_encoder = None