tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Feat: in colab, use Google drive path as dataset_path

Browse Source
This commit is contained in:
litagin02
2023-12-30 09:16:05 +09:00
parent ff8c6cd7cf
commit 6a3e1ca94c
1 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
webui_train.py
View File

@@ -11,8 +11,15 @@ from tools.log import logger
from tools.subprocess_utils import run_script_with_log, second_elem_of from tools.subprocess_utils import run_script_with_log, second_elem_of
is_colab = "google.colab" in sys.modules
def get_path(model_name): def get_path(model_name):
assert model_name != "", "モデル名は空にできません" assert model_name != "", "モデル名は空にできません"
if is_colab:
dataset_path = os.path.join(
"/content/drive/MyDrive/Style-Bert-VITS2/Data", model_name
)
dataset_path = os.path.join("Data", model_name) dataset_path = os.path.join("Data", model_name)
lbl_path = os.path.join(dataset_path, "esd.list") lbl_path = os.path.join(dataset_path, "esd.list")
train_path = os.path.join(dataset_path, "train.list") train_path = os.path.join(dataset_path, "train.list")