Skip to content

GitLab

Unverified Commit 9366c8f8 authored by Fanli Lin's avatar Fanli Lin Committed by GitHub
Browse files

fix bug in `require_accelerate_version_greater` (#9746) 

fix bug
parent e45c25d0
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/diffusers/utils/testing_utils.py
View file @ 9366c8f8
...@@ -425,7 +425,7 @@ def require_transformers_version_greater(transformers_version): ...@@ -425,7 +425,7 @@ def require_transformers_version_greater(transformers_version):
def require_accelerate_version_greater(accelerate_version): def require_accelerate_version_greater(accelerate_version):
def decorator(test_case): 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(importlib.metadata.version("accelerate")).base_version
) > version.parse(accelerate_version) ) > version.parse(accelerate_version)
return unittest.skipUnless( return unittest.skipUnless(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment