Merge pull request #35 from Disty0/main

Add OpenVINO
vladmandic · Sep 28, 2023 · 75b6c5a · 75b6c5a
2 parents b73ffdc + b537b07
commit 75b6c5a
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/scripts/system-info.py b/scripts/system-info.py
@@ -73,7 +73,17 @@ def get_user():
 
 def get_gpu():
     if not torch.cuda.is_available():
-        return {}
+        try:
+            if shared.cmd_opts.use_openvino:
+                from modules.intel.openvino import get_openvino_device
+                return {
+                    'device': get_openvino_device(),
+                    'openvino': get_package_version("openvino")
+                }
+            else:
+                return {}
+        except Exception:
+            return {}
     else:
         try:
             if torch.version.cuda: