From 7a3c1da9548a2b591fffaae7696aff620ccaf9fd Mon Sep 17 00:00:00 2001
From: Disty0 <semihgulec2005@gmail.com>
Date: Thu, 28 Sep 2023 19:33:14 +0300
Subject: [PATCH] Add OpenVINO device logging

---
 modules/devices.py                 | 15 +++++++++++++--
 modules/intel/openvino/__init__.py |  8 +++++++-
 2 files changed, 20 insertions(+), 3 deletions(-)

diff --git a/modules/devices.py b/modules/devices.py
index 797807edc..346f178ba 100644
--- a/modules/devices.py
+++ b/modules/devices.py
@@ -33,7 +33,14 @@ def get_gpu_info():
             return ''
 
     if not torch.cuda.is_available():
-        return {}
+        if hasattr(shared.cmd_opts, "use_openvino") and shared.cmd_opts.use_openvino:
+            from importlib.metadata import version as libmodule_version
+            return {
+                'device': get_openvino_device(),
+                'openvino': libmodule_version("openvino"),
+            }
+        else:
+            return {}
     else:
         try:
             if torch.version.cuda:
@@ -212,8 +219,9 @@ def set_cuda_params():
         inference_context = contextlib.nullcontext
     else:
         inference_context = torch.no_grad
+    log_device_name = get_openvino_device() if shared.cmd_opts.use_openvino else torch.device(get_optimal_device_name())
     log.debug(f'Desired Torch parameters: dtype={shared.opts.cuda_dtype} no-half={shared.opts.no_half} no-half-vae={shared.opts.no_half_vae} upscast={shared.opts.upcast_sampling}')
-    log.info(f'Setting Torch parameters: device={torch.device(get_optimal_device_name())} dtype={dtype} vae={dtype_vae} unet={dtype_unet} context={inference_context.__name__} fp16={fp16_ok} bf16={bf16_ok}')
+    log.info(f'Setting Torch parameters: device={log_device_name} dtype={dtype} vae={dtype_vae} unet={dtype_unet} context={inference_context.__name__} fp16={fp16_ok} bf16={bf16_ok}')
 
 
 args = cmd_args.parser.parse_args()
@@ -232,6 +240,9 @@ elif args.use_directml:
     if not ok:
         log.error('DirectML initialization failed: {e}')
         backend = 'cpu'
+if args.use_openvino:
+    from modules.intel.openvino import get_openvino_device
+    backend = 'openvino'
 elif torch.cuda.is_available() and torch.version.cuda:
     backend = 'cuda'
 elif torch.cuda.is_available() and torch.version.hip:
diff --git a/modules/intel/openvino/__init__.py b/modules/intel/openvino/__init__.py
index 698c162be..df61dc882 100644
--- a/modules/intel/openvino/__init__.py
+++ b/modules/intel/openvino/__init__.py
@@ -84,9 +84,15 @@ def get_device():
         device = "CPU"
         shared.log.warning("OpenVINO: No compatible GPU detected!")
     os.environ.setdefault('OPENVINO_TORCH_BACKEND_DEVICE', device)
-    shared.log.debug(f"OpenVINO Device: {device}")
     return device
 
+def get_openvino_device():
+    core = Core()
+    try:
+        return core.get_property(get_device(), "FULL_DEVICE_NAME")
+    except:
+        return "OpenVINO"
+
 def cache_root_path():
     cache_root = "./cache/"
     if os.getenv("OPENVINO_TORCH_CACHE_DIR") is not None: