github/DeepFaceLive
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLive synced 2025-07-31 04:00:12 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

refactoring

Browse source
This commit is contained in:
iperov 2021-11-07 10:03:15 +04:00
parent 8489949f2c
commit 30ba51edf7
24 changed files with 663 additions and 459 deletions
Download patch file Download diff file Expand all files Collapse all files

2
modelhub/DFLive/DFMModel.py
View file

@ -58,7 +58,7 @@ def get_available_devices() -> List[ORTDeviceInfo]:
class DFMModel:
def __init__(self, model_path : Path, device : ORTDeviceInfo = None):
if device is None:
device = lib_ort.get_cpu_device()
device = lib_ort.get_cpu_device_info()
self._model_path = model_path
sess = self._sess = lib_ort.InferenceSession_with_device(str(model_path), device)