Merge pull request #35 from Disty0/main

Add OpenVINO
pull/36/head
Vladimir Mandic 2023-09-28 12:52:43 -04:00 committed by GitHub
commit 75b6c5a6cd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 1 deletions

View File

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