apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): remove any from device pool

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

3
api/onnx_web/serve.py
View File

@ -400,7 +400,8 @@ load_platforms(context)
app = Flask(__name__)
CORS(app, origins=context.cors_origin)
executor = DevicePoolExecutor(available_platforms)
# any is a fake device, should not be in the pool
executor = DevicePoolExecutor([p for p in available_platforms if p.device != "any"])
if is_debug():
gc.set_debug(gc.DEBUG_STATS)