storytold/storyteller-ml
2
0
Fork 0
mirror of https://github.com/storytold/storyteller-ml.git synced 2026-10-09 00:09:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

log where the models are being saved

Browse Source
This commit is contained in:
Brandon Thomas
2023-05-04 04:36:37 -07:00
parent d6f4622cee
commit 7e78ec00af
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
voice_conversion/so-vits-svc-paul-v2/src/so_vits_svc_fork/utils.py
+3
View File
@@ -113,13 +113,16 @@ from joblib import Parallel, delayed
def ensure_pretrained_model(
folder_path: Path | str, type_: str, **tqdm_kwargs: Any
) -> tuple[Path, ...] | None:
LOG.info("ensure_pretrained_model: Saving pretrained model to {}".format(folder_path))
folder_path = Path(folder_path)
models_candidates = PRETRAINED_MODEL_URLS.get(type_, None)
if models_candidates is None:
LOG.warning(f"Unknown pretrained model type: {type_}")
return
for model_urls in models_candidates:
LOG.info("ensure_pretrained_model: model urls: {}".format(model_urls))
paths = [folder_path / model_url.split("/")[-1] for model_url in model_urls]
LOG.info("ensure_pretrained_model: file paths: {}".format(paths))
try:
Parallel(n_jobs=len(paths))(
[