From 9366c8f84bfe47099ff047272661786ebb54721d Mon Sep 17 00:00:00 2001 From: Fanli Lin Date: Wed, 23 Oct 2024 12:31:33 +0800 Subject: [PATCH] fix bug in `require_accelerate_version_greater` (#9746) fix bug --- src/diffusers/utils/testing_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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(