OpenVINO use the last device on the device list instead of CPU

This commit is contained in:
Disty0
2024-01-02 21:03:56 +03:00
parent 068f0a7d71
commit d7f4043093
+2 -2
View File
@@ -96,8 +96,8 @@ def get_device():
elif "GPU.0" in core.available_devices:
device = "GPU.0"
else:
device = "CPU"
shared.log.warning("OpenVINO: No compatible GPU detected!")
device = core.available_devices[-1]
shared.log.warning(f"OpenVINO: No compatible GPU detected! Using {device}")
os.environ.setdefault('OPENVINO_TORCH_BACKEND_DEVICE', device)
return device