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

feat: support use gpu

Browse Source
This commit is contained in:
sudowind
2025-04-11 08:28:25 +08:00
parent d2cb9dac2b
commit 98fcde60e8
2 changed files with 13 additions and 9 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)