Browse Source

[to #41669377] tts using default master revision model in UT

Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/10526747
master
jiaqi.sjq 3 years ago
parent
commit
5190c7de11
1 changed files with 4 additions and 7 deletions
  1. +4
    -7
      tests/pipelines/test_text_to_speech.py

+ 4
- 7
tests/pipelines/test_text_to_speech.py View File

@@ -9,7 +9,6 @@ import unittest
import torch import torch
from scipy.io.wavfile import write from scipy.io.wavfile import write


from modelscope.models import Model
from modelscope.outputs import OutputKeys from modelscope.outputs import OutputKeys
from modelscope.pipelines import pipeline from modelscope.pipelines import pipeline
from modelscope.utils.constant import Tasks from modelscope.utils.constant import Tasks
@@ -53,9 +52,8 @@ class TextToSpeechSambertHifigan16kPipelineTest(unittest.TestCase,
def test_pipeline(self): def test_pipeline(self):
for i in range(len(self.zhcn_voices)): for i in range(len(self.zhcn_voices)):
logger.info('test %s' % self.zhcn_voices[i]) logger.info('test %s' % self.zhcn_voices[i])
model = Model.from_pretrained(
model_name_or_path=self.zhcn_models[i], revision='pytorch_am')
sambert_hifigan_tts = pipeline(task=self.task, model=model)
sambert_hifigan_tts = pipeline(
task=self.task, model=self.zhcn_models[i])
self.assertTrue(sambert_hifigan_tts is not None) self.assertTrue(sambert_hifigan_tts is not None)
output = sambert_hifigan_tts(input=self.zhcn_text) output = sambert_hifigan_tts(input=self.zhcn_text)
self.assertIsNotNone(output[OutputKeys.OUTPUT_PCM]) self.assertIsNotNone(output[OutputKeys.OUTPUT_PCM])
@@ -63,9 +61,8 @@ class TextToSpeechSambertHifigan16kPipelineTest(unittest.TestCase,
write('output_%s.wav' % self.zhcn_voices[i], 16000, pcm) write('output_%s.wav' % self.zhcn_voices[i], 16000, pcm)
for i in range(len(self.en_voices)): for i in range(len(self.en_voices)):
logger.info('test %s' % self.en_voices[i]) logger.info('test %s' % self.en_voices[i])
model = Model.from_pretrained(
model_name_or_path=self.en_models[i], revision='pytorch_am')
sambert_hifigan_tts = pipeline(task=self.task, model=model)
sambert_hifigan_tts = pipeline(
task=self.task, model=self.en_models[i])
self.assertTrue(sambert_hifigan_tts is not None) self.assertTrue(sambert_hifigan_tts is not None)
output = sambert_hifigan_tts(input=self.en_text) output = sambert_hifigan_tts(input=self.en_text)
self.assertIsNotNone(output[OutputKeys.OUTPUT_PCM]) self.assertIsNotNone(output[OutputKeys.OUTPUT_PCM])


Loading…
Cancel
Save