"git@developer.sourcefind.cn:OpenDAS/vision.git" did not exist on "2b73a4846773a670632b29fb2fc2ac57df7bce5d"
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
...@@ -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