Method set_context

src/diffusers/hooks/hooks.py:49–50

(self, name: str)

        return self._state_cache[self._current_context]

    def set_context(self, name: str) -> None:
        self._current_context = name

    def reset(self, *args, **kwargs) -> None:
        for name, state in list(self._state_cache.items()):