fix(api): filter platforms based on available providers (fixes #69)

api/onnx_web/serve.py

@@ -19,6 +19,7 @@
 from glob import glob
 from io import BytesIO
 from PIL import Image
+from onnxruntime import get_available_providers
 from os import makedirs, path, scandir
 from typing import Tuple
 
@@ -72,7 +73,7 @@
     'cuda': 'CUDAExecutionProvider',
     'directml': 'DmlExecutionProvider',
     'nvidia': 'CUDAExecutionProvider',
-    'rocm': 'ROCmExecutionProvider',
+    'rocm': 'ROCMExecutionProvider',
 }
 pipeline_schedulers = {
     'ddim': DDIMScheduler,
@@ -102,6 +103,9 @@
     'gaussian-screen': mask_filter_gaussian_screen,
 }
 
+# Available ORT providers
+available_platforms = []
+
 # loaded from model_path
 diffusion_models = []
 correction_models = []
@@ -248,11 +252,21 @@ def load_params(context: ServerContext):
         config_params = json.load(f)
 
 
+def load_platforms():
+    global available_platforms
+
+    providers = get_available_providers()
+    available_platforms = [p for p in platform_providers if (platform_providers[p] in providers)]
+
+    print('available acceleration platforms: %s' % (available_platforms))
+
+
 context = ServerContext.from_environ()
 
 check_paths(context)
 load_models(context)
 load_params(context)
+load_platforms()
 
 app = Flask(__name__)
 app.config['EXECUTOR_MAX_WORKERS'] = context.num_workers