Skip to content

GitLab

Unverified Commit 0866e85e authored by daquexian's avatar daquexian Committed by GitHub
Browse files

apply_forward_hook simply returns if no accelerate (#2387) 


Signed-off-by: default avatardaquexian <daquexian566@gmail.com>
parent d2e2c611
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
src/diffusers/utils/accelerate_utils.py
View file @ 0866e85e
...@@ -34,6 +34,8 @@ def apply_forward_hook(method): ...@@ -34,6 +34,8 @@ def apply_forward_hook(method):
:param method: The method to decorate. This method should be a method of a PyTorch module. :param method: The method to decorate. This method should be a method of a PyTorch module.
""" """
if not is_accelerate_available():
return method
accelerate_version = version.parse(accelerate.__version__).base_version accelerate_version = version.parse(accelerate.__version__).base_version
if version.parse(accelerate_version) < version.parse("0.17.0"): if version.parse(accelerate_version) < version.parse("0.17.0"):
return method return method
......
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