mirror of
https://github.com/RVC-Boss/GPT-SoVITS.git
synced 2025-10-07 15:19:59 +08:00
Support V2
This commit is contained in:
parent
f13685575b
commit
abc47f14e9
@ -8,6 +8,7 @@ os.environ["CUDA_VISIBLE_DEVICES"] = os.environ.get("_CUDA_VISIBLE_DEVICES")
|
|||||||
opt_dir = os.environ.get("opt_dir")
|
opt_dir = os.environ.get("opt_dir")
|
||||||
pretrained_s2G = os.environ.get("pretrained_s2G")
|
pretrained_s2G = os.environ.get("pretrained_s2G")
|
||||||
s2config_path = os.environ.get("s2config_path")
|
s2config_path = os.environ.get("s2config_path")
|
||||||
|
version=os.environ.get("version","v2")
|
||||||
import torch
|
import torch
|
||||||
is_half = eval(os.environ.get("is_half", "True")) and torch.cuda.is_available()
|
is_half = eval(os.environ.get("is_half", "True")) and torch.cuda.is_available()
|
||||||
import math, traceback
|
import math, traceback
|
||||||
@ -50,6 +51,7 @@ if os.path.exists(semantic_path) == False:
|
|||||||
hps.data.filter_length // 2 + 1,
|
hps.data.filter_length // 2 + 1,
|
||||||
hps.train.segment_size // hps.data.hop_length,
|
hps.train.segment_size // hps.data.hop_length,
|
||||||
n_speakers=hps.data.n_speakers,
|
n_speakers=hps.data.n_speakers,
|
||||||
|
version=version,
|
||||||
**hps.model
|
**hps.model
|
||||||
)
|
)
|
||||||
if is_half == True:
|
if is_half == True:
|
||||||
|
Loading…
x
Reference in New Issue
Block a user