from local

This commit is contained in:
dohe0342 2022-12-20 15:56:42 +09:00
parent 29d534e9da
commit 78b876968b
2 changed files with 2 additions and 2 deletions

View File

@ -170,9 +170,9 @@ def download_w2v(model_url, dir_path):
if not os.path.exists(model_path):
torch.hub.download_url_to_file(model_url, model_path)
torch.hub.download_url_to_file(dict_url, dict_path)
logging.info(f"Wav2Vec model downloaded {model_path}")
logging.info(f"data2vec model downloaded {model_path}")
else:
logging.info(f"Wav2Vec model {model_path} already exists.")
logging.info(f"data2vec model {model_path} already exists.")
return model_path