Method get

src/tabpfn/inference.py:1433–1440

Return the model on the given device. Raises: KeyError: If a device is specified that was not included in the last call to .to()

(self, device: torch.device)

        self._models = {device: get_on_device(device) for device in devices}

    def get(self, device: torch.device) -> Architecture:
        """Return the model on the given device.

        Raises:
            KeyError: If a device is specified that was not included in the last call to
                .to()
        """
        return self._models[device]

    def set_dtype(self, dtype: torch.dtype) -> None:
        """Set the dtype of the model's parameters."""