fix for load specified model files

pull/2222/head
TianYuan 2 years ago
parent b9ade18055
commit cd662a08e0

@ -364,7 +364,7 @@ class TTSExecutor(BaseExecutor):
else: else:
self.am_ckpt = os.path.abspath(am_ckpt[0]) self.am_ckpt = os.path.abspath(am_ckpt[0])
self.phones_dict = os.path.abspath(phones_dict) self.phones_dict = os.path.abspath(phones_dict)
self.am_res_path = os.path.dirname(os.path.abspath(am_ckpt)) self.am_res_path = os.path.dirname(os.path.abspath(self.am_ckpt))
self.am_fs = fs self.am_fs = fs
# for speedyspeech # for speedyspeech
@ -404,7 +404,6 @@ class TTSExecutor(BaseExecutor):
# frontend # frontend
self.frontend = get_frontend( self.frontend = get_frontend(
lang=lang, phones_dict=self.phones_dict, tones_dict=self.tones_dict) lang=lang, phones_dict=self.phones_dict, tones_dict=self.tones_dict)
self.am_sess = get_sess( self.am_sess = get_sess(
model_path=self.am_ckpt, device=device, cpu_threads=cpu_threads) model_path=self.am_ckpt, device=device, cpu_threads=cpu_threads)

Loading…
Cancel
Save