diff --git a/api/onnx_web/device_pool.py b/api/onnx_web/device_pool.py index c7cd6395..ff6ca4c2 100644 --- a/api/onnx_web/device_pool.py +++ b/api/onnx_web/device_pool.py @@ -159,7 +159,7 @@ class DevicePoolExecutor: def get_next_device(self, needs_device: Optional[DeviceParams] = None) -> int: # respect overrides if possible if needs_device is not None: - for i in self.devices: + for i in range(len(self.devices)): if self.devices[i].device == needs_device.device: return i