Method from_disk

open_instruct/model_utils.py:113–115

Load a cache from disk.

(cls, path: str | pathlib.Path, device: torch.device)

    @classmethod
    def from_disk(cls, path: str | pathlib.Path, device: torch.device) -> "TensorCache":
        """Load a cache from disk."""
        return cls(tensors=torch.load(path, weights_only=True, map_location=device))


@dataclass