fix: auto backend logics

guimondmm · Mar 17, 2023 · 6b5815c · 6b5815c
1 parent 3423a33
commit 6b5815c
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/inference_realesrgan.py b/inference_realesrgan.py
@@ -109,7 +109,12 @@ def main():
 
     # deternime backend type (cpu, cuda, mps)
     if args.backend_type == 'auto':
-        backend_type = 'cuda' if cudaIsAvailable() else 'cpu'
+        if cudaIsAvailable():
+            backend_type = 'cuda'
+        elif mpsIsAvailable():
+            backend_type = 'mps'
+        else:
+            backend_type = 'cpu'
     elif args.backend_type == 'cuda' and cudaIsAvailable():
         backend_type = 'cuda'
     elif args.backend_type == 'mps' and mpsIsAvailable():