1
0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00

fix bug in require_accelerate_version_greater (#9746)

fix bug
This commit is contained in:
Fanli Lin
2024-10-23 12:31:33 +08:00
committed by GitHub
parent e45c25d03a
commit 9366c8f84b

View File

@@ -425,7 +425,7 @@ def require_transformers_version_greater(transformers_version):
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(