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

Change to manually download default models instead of auto

Browse Source
This commit is contained in:
litagin02
2024-06-02 15:31:55 +09:00
parent e11e3efce9
commit ab158d1a7b
6 changed files with 22 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
server_editor.py
View File

@@ -30,7 +30,6 @@ from pydantic import BaseModel
from scipy.io import wavfile
from config import get_path_config
from initialize import download_default_models
from style_bert_vits2.constants import (
DEFAULT_ASSIST_TEXT_WEIGHT,
DEFAULT_NOISE,
@@ -183,7 +182,7 @@ parser.add_argument("--port", type=int, default=8000)
parser.add_argument("--inbrowser", action="store_true")
parser.add_argument("--line_length", type=int, default=None)
parser.add_argument("--line_count", type=int, default=None)
parser.add_argument("--skip_default_models", action="store_true")
# parser.add_argument("--skip_default_models", action="store_true")
parser.add_argument("--skip_static_files", action="store_true")
args = parser.parse_args()
device = args.device
@@ -191,8 +190,8 @@ if device == "cuda" and not torch.cuda.is_available():
device = "cpu"
model_dir = Path(args.model_dir)
port = int(args.port)
if not args.skip_default_models:
download_default_models()
# if not args.skip_default_models:
# download_default_models()
skip_static_files = bool(args.skip_static_files)
model_holder = TTSModelHolder(model_dir, device)