apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix device iteration

This commit is contained in:
Sean Sube 2023-02-11 16:19:08 -06:00
parent db06f4a9c3
commit 66c2ce3074
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/onnx_web/device_pool.py
View File

@ -159,7 +159,7 @@ class DevicePoolExecutor:
def get_next_device(self, needs_device: Optional[DeviceParams] = None) -> int: def get_next_device(self, needs_device: Optional[DeviceParams] = None) -> int:
# respect overrides if possible # respect overrides if possible
if needs_device is not None: 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: if self.devices[i].device == needs_device.device:
return i return i