From 1ca8781bb846448df48b41feac07d85f4bd88edf Mon Sep 17 00:00:00 2001
From: kan-bayashi <hayashi.tomoki@g.sp.m.is.nagoya-u.ac.jp>
Date: Tue, 18 Aug 2020 19:55:37 +0900
Subject: [PATCH] fixed

---
 parallel_wavegan/utils/utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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]