Function _get_submodule_from_fqn

src/diffusers/hooks/_common.py:57–61

(module: torch.nn.Module, fqn: str)

def _get_submodule_from_fqn(module: torch.nn.Module, fqn: str) -> torch.nn.Module | None:
    for submodule_name, submodule in module.named_modules():
        if submodule_name == fqn:
            return submodule
    return None