MCPcopy Index your code
hub / github.com/huggingface/diffusers / _is_group_offload_enabled

Function _is_group_offload_enabled

src/diffusers/hooks/group_offloading.py:1016–1018  ·  view source on GitHub ↗
(module: torch.nn.Module)

Source from the content-addressed store, hash-verified

1014
1015
1016def _is_group_offload_enabled(module: torch.nn.Module) -> bool:
1017 top_level_group_offload_hook = _get_top_level_group_offload_hook(module)
1018 return top_level_group_offload_hook is not None
1019
1020
1021def _get_group_onload_device(module: torch.nn.Module) -> torch.device:

Callers 5

_maybe_raise_error_if_group_offload_activeMethod · 0.85
_func_optionally_disable_offloadingFunction · 0.85
_maybe_raise_error_if_group_offload_activeMethod · 0.85
cudaMethod · 0.85
toMethod · 0.85

Calls 1

_get_top_level_group_offload_hookFunction · 0.85

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…