diff --git a/api/onnx_web/serve.py b/api/onnx_web/serve.py index fd9ebd0f..419070d2 100644 --- a/api/onnx_web/serve.py +++ b/api/onnx_web/serve.py @@ -84,11 +84,9 @@ config_params = {} # pipeline params platform_providers = { - "amd": "DmlExecutionProvider", "cpu": "CPUExecutionProvider", "cuda": "CUDAExecutionProvider", "directml": "DmlExecutionProvider", - "nvidia": "CUDAExecutionProvider", "rocm": "ROCMExecutionProvider", } pipeline_schedulers = { diff --git a/onnx-web.code-workspace b/onnx-web.code-workspace index 20d949ba..78337b97 100644 --- a/onnx-web.code-workspace +++ b/onnx-web.code-workspace @@ -42,6 +42,7 @@ "numpy", "Onnx", "onnxruntime", + "openjourney", "opset", "outpaint", "outpainting",