Unverified Commit bbd0c161 authored by Marc Sun's avatar Marc Sun Committed by GitHub
Browse files

[BNB] Fix test_moving_to_cpu_throws_warning (#11356)



fix
Co-authored-by: default avatarSayak Paul <spsayakpaul@gmail.com>
parent eef3d659
...@@ -404,6 +404,11 @@ class DiffusionPipeline(ConfigMixin, PushToHubMixin): ...@@ -404,6 +404,11 @@ class DiffusionPipeline(ConfigMixin, PushToHubMixin):
if not is_accelerate_available() or is_accelerate_version("<", "0.14.0"): if not is_accelerate_available() or is_accelerate_version("<", "0.14.0"):
return False 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 ( return hasattr(module, "_hf_hook") and (
isinstance(module._hf_hook, accelerate.hooks.AlignDevicesHook) isinstance(module._hf_hook, accelerate.hooks.AlignDevicesHook)
or hasattr(module._hf_hook, "hooks") or hasattr(module._hf_hook, "hooks")
......
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