[model][NPU]:Wan model rope use torch.complex64 in NPU

This commit is contained in:
feng0w0
2026-01-12 11:34:41 +08:00
parent 544c391936
commit 6be244233a

View File

@@ -1,2 +1,2 @@
from .npu_compatible_device import parse_device_type, parse_nccl_backend, get_available_device_type, get_device_name
from .npu_compatible_device import IS_NPU_AVAILABLE
from .npu_compatible_device import IS_NPU_AVAILABLE, IS_CUDA_AVAILABLE