Method cache_context

src/diffusers/models/cache_utils.py:155–164

r"""Context manager that provides additional methods for cache management.

(self, name: str)

    @contextmanager
    def cache_context(self, name: str):
        r"""Context manager that provides additional methods for cache management."""
        from ..hooks import HookRegistry

        registry = HookRegistry.check_if_exists_or_initialize(self)
        registry._set_context(name)

        yield

        registry._set_context(None)