def get_output_models_by_device(self, device: Union[Device, str]) -> Optional[list[ModelOutput]]:
        """Get the output model for a specific device.

        Args:
            device: The device to retrieve the output model for

        """
        device_type = str(device).lower()
        if device_type not in self._device_outputs:
            return None
        return self._device_outputs.get(device_type).get_output_models()
Get the output model for a specific device.

        Args:
            device: The device to retrieve the output model for

        
olive/engine/output.py