Method _set_context

src/diffusers/hooks/hooks.py:268–275

(self, name: str | None = None)

        return module._diffusers_hook

    def _set_context(self, name: str | None = None) -> None:
        for hook_name in reversed(self._hook_order):
            hook = self.hooks[hook_name]
            if hook._is_stateful:
                hook._set_context(self._module_ref, name)

        for registry in self._get_child_registries():
            registry._set_context(name)

    def _get_child_registries(self) -> list["HookRegistry"]:
        """Return registries of child modules, using a cached list when available.