From 0866e85e76f6d41d128f49ae0aee2c7bf74dbd95 Mon Sep 17 00:00:00 2001 From: daquexian Date: Sat, 18 Feb 2023 00:27:23 +0800 Subject: [PATCH] apply_forward_hook simply returns if no accelerate (#2387) Signed-off-by: daquexian --- src/diffusers/utils/accelerate_utils.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/diffusers/utils/accelerate_utils.py b/src/diffusers/utils/accelerate_utils.py index a92d475476..10a83e1dd2 100644 --- a/src/diffusers/utils/accelerate_utils.py +++ b/src/diffusers/utils/accelerate_utils.py @@ -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. """ + if not is_accelerate_available(): + return method accelerate_version = version.parse(accelerate.__version__).base_version if version.parse(accelerate_version) < version.parse("0.17.0"): return method