diff --git a/vllm_omni/platforms/npu/platform.py b/vllm_omni/platforms/npu/platform.py index c40dd6fea18..53ffe6775a5 100644 --- a/vllm_omni/platforms/npu/platform.py +++ b/vllm_omni/platforms/npu/platform.py @@ -69,6 +69,9 @@ def get_diffusion_attn_backend_cls( # Try FLASH_ATTN if mindiesd is available, otherwise fall back to SDPA if find_spec("mindiesd"): + # Configure ASCEND_CUSTOM_OPP_PATH for mindiesd custom ops upon import + import mindiesd # noqa: F401 + logger.info("Defaulting to diffusion attention backend FLASH_ATTN") return DiffusionAttentionBackendEnum.FLASH_ATTN.get_path()