Method device

src/transformers/modeling_utils.py:906–911

`torch.device`: The device on which the module is (assuming that all the module parameters are on the same device).

(self: "PreTrainedModel")

    @property
    def device(self: "PreTrainedModel") -> torch.device:
        """
        `torch.device`: The device on which the module is (assuming that all the module parameters are on the same
        device).
        """
        return next(param.device for param in self.parameters())

    @property
    def dtype(self: "PreTrainedModel") -> torch.dtype: