Function unwrap_module

src/diffusers/utils/torch_utils.py:209–211

Unwraps a module if it was compiled with torch.compile()

(module)

def unwrap_module(module):
    """Unwraps a module if it was compiled with torch.compile()"""
    return module._orig_mod if is_compiled_module(module) else module


def fourier_filter(x_in: "torch.Tensor", threshold: int, scale: int) -> "torch.Tensor":