Using Wget for new ckpt downloadsA

This commit is contained in:
Shivam Mehta
2024-01-12 11:09:25 +00:00
parent 95ec24b599
commit 39cbd85236
4 changed files with 17 additions and 15 deletions

View File

@@ -74,7 +74,7 @@ def assert_required_models_available(args):
model_path = args.checkpoint_path
else:
model_path = save_dir / f"{args.model}.ckpt"
assert_model_downloaded(model_path, MATCHA_URLS[args.model])
assert_model_downloaded(model_path, MATCHA_URLS[args.model], use_wget=True)
vocoder_path = save_dir / f"{args.vocoder}"
assert_model_downloaded(vocoder_path, VOCODER_URLS[args.vocoder])