Function is_compiled_module

src/diffusers/utils/torch_utils.py:202–206

Check whether the module was compiled with torch.compile()

(module)

def is_compiled_module(module) -> bool:
    """Check whether the module was compiled with torch.compile()"""
    if is_torch_version("<", "2.0.0") or not hasattr(torch, "_dynamo"):
        return False
    return isinstance(module, torch._dynamo.eval_frame.OptimizedModule)


def unwrap_module(module):