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

control hubert loading/saving

Browse Source
This commit is contained in:
Brandon Thomas
2023-04-28 15:26:07 -07:00
parent c43c5ae99c
commit 044883d3e5
1 changed files with 12 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
voice_conversion/so-vits-svc-paul/src/so_vits_svc_fork/utils.py
+12 -1
View File
@@ -1,5 +1,6 @@
from __future__ import annotations from __future__ import annotations
import os
import json import json
import re import re
from itertools import groupby from itertools import groupby
@@ -312,9 +313,19 @@ def ensure_pretrained_model(folder_path: Path) -> None:
if not model_path.exists(): if not model_path.exists():
download_file(model_url, model_path, desc=f"Downloading {model_path.name}") download_file(model_url, model_path, desc=f"Downloading {model_path.name}")
def fakeyou_hubert_path():
"""
Env var `HUBERT_PATH` lets external callers set where Hubert gets loaded or saved.
"""
default_path = 'checkpoint_best_legacy_500.pt' # current directory
hubert_path = os.getenv('HUBERT_PATH')
if not hubert_path:
hubert_path = default_path
return Path(hubert_path)
def ensure_hubert_model() -> Path: def ensure_hubert_model() -> Path:
vec_path = Path("checkpoint_best_legacy_500.pt") # NB(bt): We need to control where hubert gets loaded from / saved to.
vec_path = fakeyou_hubert_path()
vec_path.parent.mkdir(parents=True, exist_ok=True) vec_path.parent.mkdir(parents=True, exist_ok=True)
if not vec_path.exists(): if not vec_path.exists():
# url = "http://obs.cstcloud.cn/share/obs/sankagenkeshi/checkpoint_best_legacy_500.pt" # url = "http://obs.cstcloud.cn/share/obs/sankagenkeshi/checkpoint_best_legacy_500.pt"