Function _is_torchao_tensor

src/diffusers/hooks/group_offloading.py:38–43

(tensor: torch.Tensor)

def _is_torchao_tensor(tensor: torch.Tensor) -> bool:
    if not is_torchao_available():
        return False
    from torchao.utils import TorchAOBaseTensor

    return isinstance(tensor, TorchAOBaseTensor)


def _get_torchao_inner_tensor_names(tensor: torch.Tensor) -> list[str]: