This commit is contained in:
Disty0
2023-09-28 19:53:52 +03:00
parent 7a3c1da954
commit 21d53b6ac8
2 changed files with 11 additions and 8 deletions
+10 -7
View File
@@ -33,13 +33,16 @@ def get_gpu_info():
return ''
if not torch.cuda.is_available():
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:
try:
if shared.cmd_opts.use_openvino:
from importlib.metadata import version as libmodule_version
return {
'device': get_openvino_device(),
'openvino': libmodule_version("openvino"),
}
else:
return {}
except Exception:
return {}
else:
try:
+1 -1
View File
@@ -90,7 +90,7 @@ def get_openvino_device():
core = Core()
try:
return core.get_property(get_device(), "FULL_DEVICE_NAME")
except:
except Exception:
return "OpenVINO"
def cache_root_path():