From 84cbcf0888c61b51713b9db1cffbef045a69a9b3 Mon Sep 17 00:00:00 2001 From: Mikko Lauri <167742591+lauri9@users.noreply.github.com> Date: Fri, 30 Jan 2026 14:36:18 +0200 Subject: [PATCH] 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 425c360a3110..e35fc9697076 100644 --- a/src/diffusers/utils/import_utils.py +++ b/src/diffusers/utils/import_utils.py @@ -227,7 +227,7 @@ def _is_package_available(pkg_name: str, get_dist_name: bool = False) -> Tuple[b _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")