Function patch_context

diffusers_osinsert/__init__.py:67–73

Temporarily enable the patch within a `with` block and restore afterwards.

()

@contextmanager
def patch_context():
    """Temporarily enable the patch within a `with` block and restore afterwards."""
    enable_patch()
    try:
        yield
    finally:
        disable_patch()


# Re-export upstream classes for convenience.