Function _apply_fbc_block_hook

src/diffusers/hooks/first_block_cache.py:255–258

(block: torch.nn.Module, state_manager: StateManager, is_tail: bool = False)

def _apply_fbc_block_hook(block: torch.nn.Module, state_manager: StateManager, is_tail: bool = False) -> None:
    registry = HookRegistry.check_if_exists_or_initialize(block)
    hook = FBCBlockHook(state_manager, is_tail)
    registry.register_hook(hook, _FBC_BLOCK_HOOK)