bacow/GPT-SoVITS
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix cpu+fp16 inference issue

Browse Source 
fix cpu+fp16 inference issue
This commit is contained in:
RVC-Boss
2024-08-01 21:47:30 +08:00
committed by GitHub
parent 9498fc775b
commit bce451a2d1
3 changed files with 8 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
GPT_SoVITS/prepare_datasets/3-get-semantic.py
View File

@@ -8,7 +8,8 @@ os.environ["CUDA_VISIBLE_DEVICES"] = os.environ.get("_CUDA_VISIBLE_DEVICES")
opt_dir = os.environ.get("opt_dir")
pretrained_s2G = os.environ.get("pretrained_s2G")
s2config_path = os.environ.get("s2config_path")
is_half = eval(os.environ.get("is_half", "True"))
import torch
is_half = eval(os.environ.get("is_half", "True")) and torch.cuda.is_available()
import math, traceback
import multiprocessing
import sys, pdb
@@ -19,7 +20,7 @@ from random import shuffle
import torch.multiprocessing as mp
from glob import glob
from tqdm import tqdm
import logging, librosa, utils, torch
import logging, librosa, utils
from module.models import SynthesizerTrn
logging.getLogger("numba").setLevel(logging.WARNING)