def to()

in optimum/furiosa/modeling.py [0:0]

• 4 lines of code
• 1 McCabe index (conditional complexity)

    def to(self, device: str):
        """
        Use the specified `device` for inference. For example: "cpu" or "gpu". `device` can
        be in upper or lower case. To speed up first inference, call `.compile()` after `.to()`.
        """
        self._device = device.upper()
        self.sess = None
        return self