Method _to_cpu

src/diffusers/hooks/group_offloading.py:176–180

(tensor, low_cpu_mem_usage)

    @staticmethod
    def _to_cpu(tensor, low_cpu_mem_usage):
        # For TorchAO tensors, `.data` returns an incomplete wrapper without internal attributes
        # (e.g. `.qdata`, `.scale`), so we must call `.cpu()` on the tensor directly.
        t = tensor.cpu() if _is_torchao_tensor(tensor) else tensor.data.cpu()
        return t if low_cpu_mem_usage else t.pin_memory()

    def _init_cpu_param_dict(self):
        cpu_param_dict = {}