From ec6b2bcccb650ca94a0a1d98fb9e96941d5d188f Mon Sep 17 00:00:00 2001 From: Mikko Lauri <167742591+lauri9@users.noreply.github.com> Date: Fri, 30 Jan 2026 15:54:05 +0200 Subject: [PATCH] Fix aiter availability check (#13059) Update import_utils.py --- src/diffusers/utils/import_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/utils/import_utils.py b/src/diffusers/utils/import_utils.py index 425c360a31..e35fc96970 100644 --- a/src/diffusers/utils/import_utils.py +++ b/src/diffusers/utils/import_utils.py @@ -227,7 +227,7 @@ _cosmos_guardrail_available, _cosmos_guardrail_version = _is_package_available(" _sageattention_available, _sageattention_version = _is_package_available("sageattention") _flash_attn_available, _flash_attn_version = _is_package_available("flash_attn") _flash_attn_3_available, _flash_attn_3_version = _is_package_available("flash_attn_3") -_aiter_available, _aiter_version = _is_package_available("aiter") +_aiter_available, _aiter_version = _is_package_available("aiter", get_dist_name=True) _kornia_available, _kornia_version = _is_package_available("kornia") _nvidia_modelopt_available, _nvidia_modelopt_version = _is_package_available("modelopt", get_dist_name=True) _av_available, _av_version = _is_package_available("av")