Change to manually download default models instead of auto
This commit is contained in:
7
app.py
7
app.py
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user