diff --git a/server/voice_changer/VoiceChanger.py b/server/voice_changer/VoiceChanger.py index cc17e039..72f77a19 100755 --- a/server/voice_changer/VoiceChanger.py +++ b/server/voice_changer/VoiceChanger.py @@ -105,6 +105,8 @@ class VoiceChanger(): def update_setteings(self, key:str, val:any): if key == "onnxExecutionProvider" and self.onnx_session != None: if val == "CUDAExecutionProvider": + if self.settings.gpu < 0 or self.settings.gpu >= self.gpu_num: + self.settings.gpu = 0 provider_options=[{'device_id': self.settings.gpu}] self.onnx_session.set_providers(providers=[val], provider_options=provider_options) else: @@ -115,8 +117,6 @@ class VoiceChanger(): providers = self.onnx_session.get_providers() print("Providers:", providers) if "CUDAExecutionProvider" in providers: - if self.settings.gpu < 0 or self.settings.gpu >= self.gpu_num: - self.settings.gpu = 0 provider_options=[{'device_id': self.settings.gpu}] self.onnx_session.set_providers(providers=["CUDAExecutionProvider"], provider_options=provider_options) if key == "crossFadeOffsetRate" or key == "crossFadeEndRate":