Skip to content

Commit ec6b2bc

Browse files
authored
Fix aiter availability check (#13059)
Update import_utils.py
1 parent 6a1904e commit ec6b2bc

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/utils/import_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -227,7 +227,7 @@ def _is_package_available(pkg_name: str, get_dist_name: bool = False) -> Tuple[b
227227
_sageattention_available, _sageattention_version = _is_package_available("sageattention")
228228
_flash_attn_available, _flash_attn_version = _is_package_available("flash_attn")
229229
_flash_attn_3_available, _flash_attn_3_version = _is_package_available("flash_attn_3")
230-
_aiter_available, _aiter_version = _is_package_available("aiter")
230+
_aiter_available, _aiter_version = _is_package_available("aiter", get_dist_name=True)
231231
_kornia_available, _kornia_version = _is_package_available("kornia")
232232
_nvidia_modelopt_available, _nvidia_modelopt_version = _is_package_available("modelopt", get_dist_name=True)
233233
_av_available, _av_version = _is_package_available("av")

0 commit comments

Comments
 (0)