blackteay/GPT-SoVITS
1
0
Fork 0
mirror of https://github.com/RVC-Boss/GPT-SoVITS.git synced 2025-08-17 23:19:47 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix inference process with newer torch

Browse Source
This commit is contained in:
Jarod Mica 2025-06-02 14:01:57 -07:00
parent 651b0075ad
commit 5ca23e2c70
2 changed files with 20 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
GPT_SoVITS/TTS_infer_pack/TTS.py
View File

@ -583,11 +583,18 @@ class TTS:
self.vocoder.cpu()
del self.vocoder
self.empty_cache()
self.vocoder = BigVGAN.from_pretrained(
"%s/GPT_SoVITS/pretrained_models/models--nvidia--bigvgan_v2_24khz_100band_256x" % (now_dir,),
use_cuda_kernel=False,
) # if True, RuntimeError: Ninja is required to load C++ extensions
if not vocoder_path:
self.vocoder = BigVGAN.from_pretrained(
"%s/GPT_SoVITS/pretrained_models/models--nvidia--bigvgan_v2_24khz_100band_256x" % (now_dir,),
use_cuda_kernel=False,
) # if True, RuntimeError: Ninja is required to load C++ extensions
else:
self.vocoder = BigVGAN.from_pretrained(
vocoder_path,
use_cuda_kernel=False,
local_files_only=True,
cache_dir=vocoder_path
)
# remove weight norm in the model and set to eval mode
self.vocoder.remove_weight_norm()

10
GPT_SoVITS/process_ckpt.py
View File

@ -4,6 +4,12 @@ from time import time as ttime
import shutil
import os
import torch
import sys
import GPT_SoVITS.utils as _gpt_utils
sys.modules['utils'] = _gpt_utils
from GPT_SoVITS.utils import HParams
HParams.__module__ = 'utils'
torch.serialization.add_safe_globals([HParams])
from GPT_SoVITS.tools.i18n.i18n import I18nAuto
i18n = I18nAuto()
@ -120,5 +126,5 @@ def load_sovits_new(sovits_path):
bio = BytesIO()
bio.write(data)
bio.seek(0)
return torch.load(bio, map_location="cpu", weights_only=False)
return torch.load(sovits_path, map_location="cpu", weights_only=False)
return torch.load(bio, map_location="cpu", weights_only=True)
return torch.load(sovits_path, map_location="cpu", weights_only=True)