diff --git a/diffsynth/pipelines/wan_video.py b/diffsynth/pipelines/wan_video.py index 45ea43c..5b4c0b4 100644 --- a/diffsynth/pipelines/wan_video.py +++ b/diffsynth/pipelines/wan_video.py @@ -126,8 +126,8 @@ class WanVideoPipeline(BasePipeline): from ..utils.xfuser import initialize_usp initialize_usp(device) import torch.distributed as dist - from ..core.device.npu_compatible_device import get_device_name, IS_NPU_AVAILABLE, IS_CUDA_AVAILABLE - if dist.is_available() and dist.is_initialized() and (IS_CUDA_AVAILABLE or IS_NPU_AVAILABLE): + from ..core.device.npu_compatible_device import get_device_name + if dist.is_available() and dist.is_initialized(): device = get_device_name() # Initialize pipeline pipe = WanVideoPipeline(device=device, torch_dtype=torch_dtype)