Compare commits

...

2 Commits

Author SHA1 Message Date
Sayak Paul
4209647dc8 Merge branch 'main' into fix-bnb-test 2025-04-18 08:55:15 +05:30
Marc Sun
c1a2a9d405 fix 2025-04-17 18:00:25 +02:00

View File

@@ -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")