From 6be244233a5706d0cf7e0fc8f019566f8f0dca8f Mon Sep 17 00:00:00 2001 From: feng0w0 Date: Mon, 12 Jan 2026 11:34:41 +0800 Subject: [PATCH] [model][NPU]:Wan model rope use torch.complex64 in NPU --- diffsynth/core/device/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/diffsynth/core/device/__init__.py b/diffsynth/core/device/__init__.py index 8373471..889d682 100644 --- a/diffsynth/core/device/__init__.py +++ b/diffsynth/core/device/__init__.py @@ -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