diff --git a/src/diffusers/utils/testing_utils.py b/src/diffusers/utils/testing_utils.py index 1179b113d636..6361cca663b9 100644 --- a/src/diffusers/utils/testing_utils.py +++ b/src/diffusers/utils/testing_utils.py @@ -425,7 +425,7 @@ def decorator(test_case): def require_accelerate_version_greater(accelerate_version): def decorator(test_case): - correct_accelerate_version = is_peft_available() and version.parse( + correct_accelerate_version = is_accelerate_available() and version.parse( version.parse(importlib.metadata.version("accelerate")).base_version ) > version.parse(accelerate_version) return unittest.skipUnless(