diff --git a/parallel_wavegan/utils/utils.py b/parallel_wavegan/utils/utils.py index bc21af03..65314679 100644 --- a/parallel_wavegan/utils/utils.py +++ b/parallel_wavegan/utils/utils.py @@ -337,6 +337,6 @@ def download_pretrained_model(tag, download_dir=None): gdown.download(f"https://drive.google.com/uc?id={id_}", output_path, quiet=False) with tarfile.open(output_path, 'r:*') as tar: tar.extractall(f"{download_dir}/{tag}") - checkpoint_path = find_files(f"{download_dir}/{tag}", "*.pkl") + checkpoint_path = find_files(f"{download_dir}/{tag}", "checkpoint*.pkl") return checkpoint_path[0]