From c1a2a9d40513f2dedc94da6408611cbdf7608b50 Mon Sep 17 00:00:00 2001 From: Marc Sun Date: Thu, 17 Apr 2025 18:00:25 +0200 Subject: [PATCH] fix --- src/diffusers/pipelines/pipeline_utils.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/src/diffusers/pipelines/pipeline_utils.py b/src/diffusers/pipelines/pipeline_utils.py index 381def9501..08087acd36 100644 --- a/src/diffusers/pipelines/pipeline_utils.py +++ b/src/diffusers/pipelines/pipeline_utils.py @@ -404,6 +404,11 @@ class DiffusionPipeline(ConfigMixin, PushToHubMixin): if not is_accelerate_available() or is_accelerate_version("<", "0.14.0"): return False + _, _, is_loaded_in_8bit_bnb = _check_bnb_status(module) + + if is_loaded_in_8bit_bnb: + return False + return hasattr(module, "_hf_hook") and ( isinstance(module._hf_hook, accelerate.hooks.AlignDevicesHook) or hasattr(module._hf_hook, "hooks")