Change to manually download default models instead of auto

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

7
app.py
View File

@@ -10,7 +10,6 @@ from gradio_tabs.inference import create_inference_app
from gradio_tabs.merge import create_merge_app
from gradio_tabs.style_vectors import create_style_vectors_app
from gradio_tabs.train import create_train_app
from initialize import download_default_models
from style_bert_vits2.constants import GRADIO_THEME, VERSION
from style_bert_vits2.nlp.japanese import pyopenjtalk_worker
from style_bert_vits2.nlp.japanese.user_dict import update_dict
@@ -30,15 +29,15 @@ parser.add_argument("--host", type=str, default="127.0.0.1")
parser.add_argument("--port", type=int, default=None)
parser.add_argument("--no_autolaunch", action="store_true")
parser.add_argument("--share", action="store_true")
parser.add_argument("--skip_default_models", action="store_true")
# parser.add_argument("--skip_default_models", action="store_true")
args = parser.parse_args()
device = args.device
if device == "cuda" and not torch.cuda.is_available():
device = "cpu"
if not args.skip_default_models:
download_default_models()
# if not args.skip_default_models:
# download_default_models()
path_config = get_path_config()
model_holder = TTSModelHolder(Path(path_config.assets_root), device)