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

Remove warnings and add some warnings (#1408)

Browse Source
This commit is contained in:
XXXXRT666
2024-08-06 18:37:55 +08:00
committed by GitHub
parent 2196b2bfc6
commit 805dbd7ff2
5 changed files with 61 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
GPT_SoVITS/prepare_datasets/1-get-text.py
View File

@@ -54,6 +54,8 @@ if os.path.exists(txt_path) == False:
# device = "mps"
else:
device = "cpu"
if os.path.exists(bert_pretrained_dir):...
else:raise FileNotFoundError(bert_pretrained_dir)
tokenizer = AutoTokenizer.from_pretrained(bert_pretrained_dir)
bert_model = AutoModelForMaskedLM.from_pretrained(bert_pretrained_dir)
if is_half == True:

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

@@ -34,6 +34,8 @@ logging.getLogger("numba").setLevel(logging.WARNING)
# os.environ["CUDA_VISIBLE_DEVICES"]=sys.argv[5]
# opt_dir="/data/docker/liujing04/gpt-vits/fine_tune_dataset/%s"%exp_name
if os.path.exists(pretrained_s2G):...
else:raise FileNotFoundError(pretrained_s2G)
hubert_dir = "%s/4-cnhubert" % (opt_dir)
semantic_path = "%s/6-name2semantic-%s.tsv" % (opt_dir, i_part)