Move OpenVINO in try except

pull/35/head
Disty0 2023-09-28 19:48:42 +03:00
parent 111523c163
commit b537b071b2
1 changed files with 10 additions and 7 deletions

View File

@ -73,13 +73,16 @@ def get_user():
def get_gpu():
if not torch.cuda.is_available():
if hasattr(shared.cmd_opts, "use_openvino") and shared.cmd_opts.use_openvino:
from modules.intel.openvino import get_openvino_device
return {
'device': get_openvino_device(),
'openvino': get_package_version("openvino")
}
else:
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: