From 527744d5f0ab6b91c6e0e338db091161d14534e1 Mon Sep 17 00:00:00 2001
From: KP <109694228@qq.com>
Date: Wed, 29 Jun 2022 18:06:49 +0800
Subject: [PATCH] Fix unnecessary download present in issue #2067.

---
 .../engine/tts/online/python/tts_engine.py       | 16 ++++++++++++++--
 .../engine/tts/paddleinference/tts_engine.py     | 16 ++++++++++++++--
 2 files changed, 28 insertions(+), 4 deletions(-)

diff --git a/paddlespeech/server/engine/tts/online/python/tts_engine.py b/paddlespeech/server/engine/tts/online/python/tts_engine.py
index 2e8997e0f..743643828 100644
--- a/paddlespeech/server/engine/tts/online/python/tts_engine.py
+++ b/paddlespeech/server/engine/tts/online/python/tts_engine.py
@@ -105,13 +105,19 @@ class TTSServerExecutor(TTSExecutor):
             logger.info('Models had been initialized.')
             return
         # am model info
+        if am_ckpt is None or am_config is None or am_stat is None or phones_dict is None:
+            use_pretrained_am = True
+        else:
+            use_pretrained_am = False
+
         am_tag = am + '-' + lang
         self.task_resource.set_task_model(
             model_tag=am_tag,
             model_type=0,  # am
+            skip_download=not use_pretrained_am,
             version=None,  # default version
         )
-        if am_ckpt is None or am_config is None or am_stat is None or phones_dict is None:
+        if use_pretrained_am:
             self.am_res_path = self.task_resource.res_dir
             self.am_config = os.path.join(self.am_res_path,
                                           self.task_resource.res_dict['config'])
@@ -138,13 +144,19 @@ class TTSServerExecutor(TTSExecutor):
         self.speaker_dict = None
 
         # voc model info
+        if voc_ckpt is None or voc_config is None or voc_stat is None:
+            use_pretrained_voc = True
+        else:
+            use_pretrained_voc = False
+
         voc_tag = voc + '-' + lang
         self.task_resource.set_task_model(
             model_tag=voc_tag,
             model_type=1,  # vocoder
+            skip_download=not use_pretrained_voc,
             version=None,  # default version
         )
-        if voc_ckpt is None or voc_config is None or voc_stat is None:
+        if use_pretrained_voc:
             self.voc_res_path = self.task_resource.voc_res_dir
             self.voc_config = os.path.join(
                 self.voc_res_path, self.task_resource.voc_res_dict['config'])
diff --git a/paddlespeech/server/engine/tts/paddleinference/tts_engine.py b/paddlespeech/server/engine/tts/paddleinference/tts_engine.py
index ab5b721ff..a56324871 100644
--- a/paddlespeech/server/engine/tts/paddleinference/tts_engine.py
+++ b/paddlespeech/server/engine/tts/paddleinference/tts_engine.py
@@ -68,13 +68,19 @@ class TTSServerExecutor(TTSExecutor):
             logger.info('Models had been initialized.')
             return
         # am
+        if am_model is None or am_params is None or phones_dict is None:
+            use_pretrained_am = True
+        else:
+            use_pretrained_am = False
+
         am_tag = am + '-' + lang
         self.task_resource.set_task_model(
             model_tag=am_tag,
             model_type=0,  # am
+            skip_download=not use_pretrained_am,
             version=None,  # default version
         )
-        if am_model is None or am_params is None or phones_dict is None:
+        if use_pretrained_am:
             self.am_res_path = self.task_resource.res_dir
             self.am_model = os.path.join(self.am_res_path,
                                          self.task_resource.res_dict['model'])
@@ -113,13 +119,19 @@ class TTSServerExecutor(TTSExecutor):
                 self.speaker_dict = speaker_dict
 
         # voc
+        if voc_model is None or voc_params is None:
+            use_pretrained_voc = True
+        else:
+            use_pretrained_voc = False
+
         voc_tag = voc + '-' + lang
         self.task_resource.set_task_model(
             model_tag=voc_tag,
             model_type=1,  # vocoder
+            skip_download=not use_pretrained_voc,
             version=None,  # default version
         )
-        if voc_model is None or voc_params is None:
+        if use_pretrained_voc:
             self.voc_res_path = self.task_resource.voc_res_dir
             self.voc_model = os.path.join(
                 self.voc_res_path, self.task_resource.voc_res_dict['model'])