deepgeek/lite-avatar
2
0
Fork 0
mirror of https://github.com/HumanAIGC/lite-avatar.git synced 2026-02-04 09:29:19 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update audio2mouth_cpu.py

Browse Source
This commit is contained in:
sudowind
2025-04-11 09:38:26 +08:00
committed by GitHub
parent 48d2674006
commit 7ed7dc549d
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
audio2mouth_cpu.py
View File

@@ -8,12 +8,13 @@ from extract_paraformer_feature import extract_para_feature
from scipy import signal
class Audio2Mouth(object):
def __init__(self):
def __init__(self. use_gpu):
self.p_list = [str(ii) for ii in range(32)]
model_path = './weights/model_1.onnx'
self.audio2mouth_model=onnxruntime.InferenceSession(model_path, providers=['CPUExecutionProvider'])
provider = "CUDAExecutionProvider" if use_gpu else "CPUExecutionProvider"
self.audio2mouth_model=onnxruntime.InferenceSession(model_path, providers=[provider])
self.w = np.array([1.0]).astype(np.float32)
self.sp = np.array([2]).astype(np.int64)