1berry/wokada-voice-changer
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Improve: show onnx device

Browse Source
This commit is contained in:
w-okada 2023-07-08 19:07:41 +09:00
parent bde95029b3
commit 6fb00c7d91
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
server/voice_changer/VoiceChanger.py
View File

@ -7,7 +7,7 @@ import traceback
import numpy as np import numpy as np
from dataclasses import dataclass, asdict, field from dataclasses import dataclass, asdict, field
import resampy import resampy
import onnxruntime
from voice_changer.IORecorder import IORecorder from voice_changer.IORecorder import IORecorder
@ -65,7 +65,6 @@ class VoiceChanger:
def __init__(self, params: VoiceChangerParams): def __init__(self, params: VoiceChangerParams):
# 初期化 # 初期化
self.settings = VoiceChangerSettings() self.settings = VoiceChangerSettings()
self.onnx_session = None
self.currentCrossFadeOffsetRate = 0.0 self.currentCrossFadeOffsetRate = 0.0
self.currentCrossFadeEndRate = 0.0 self.currentCrossFadeEndRate = 0.0
self.currentCrossFadeOverlapSize = 0 # setting self.currentCrossFadeOverlapSize = 0 # setting
@ -77,8 +76,9 @@ class VoiceChanger:
self.gpu_num = torch.cuda.device_count() self.gpu_num = torch.cuda.device_count()
self.prev_audio = np.zeros(4096) self.prev_audio = np.zeros(4096)
self.mps_enabled: bool = getattr(torch.backends, "mps", None) is not None and torch.backends.mps.is_available() self.mps_enabled: bool = getattr(torch.backends, "mps", None) is not None and torch.backends.mps.is_available()
self.onnx_device = onnxruntime.get_device()
print(f"VoiceChanger Initialized (GPU_NUM:{self.gpu_num}, mps_enabled:{self.mps_enabled})") print(f"VoiceChanger Initialized (GPU_NUM(cuda):{self.gpu_num}, mps_enabled:{self.mps_enabled}, onnx_device:{self.onnx_device})")
def setModel(self, model: Any): def setModel(self, model: Any):
self.voiceChanger = model self.voiceChanger = model