Add OpenVINO

pull/35/head
Disty0 2023-09-28 19:29:31 +03:00
parent b73ffdcc8d
commit 111523c163
1 changed files with 8 additions and 1 deletions

View File

@ -73,7 +73,14 @@ def get_user():
def get_gpu():
if not torch.cuda.is_available():
return {}
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:
return {}
else:
try:
if torch.version.cuda: