Move OpenVINO in try except
parent
111523c163
commit
b537b071b2
|
|
@ -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:
|
||||
|
|
|
|||
Loading…
Reference in New Issue