mirror of https://github.com/coqui-ai/TTS.git
pylint fixes and add missing requirement
This commit is contained in:
parent
b2cc256dab
commit
664f42df33
|
@ -15,6 +15,7 @@ tqdm
|
||||||
inflect
|
inflect
|
||||||
pysbd
|
pysbd
|
||||||
bokeh==1.4.0
|
bokeh==1.4.0
|
||||||
|
pysbd
|
||||||
soundfile
|
soundfile
|
||||||
nose==1.3.7
|
nose==1.3.7
|
||||||
cardboardlint==1.3.0
|
cardboardlint==1.3.0
|
||||||
|
|
|
@ -1,5 +1,4 @@
|
||||||
import io
|
import io
|
||||||
import re
|
|
||||||
import sys
|
import sys
|
||||||
import time
|
import time
|
||||||
|
|
||||||
|
|
1
setup.py
1
setup.py
|
@ -93,6 +93,7 @@ requirements = {
|
||||||
"inflect",
|
"inflect",
|
||||||
"pysbd",
|
"pysbd",
|
||||||
"bokeh==1.4.0",
|
"bokeh==1.4.0",
|
||||||
|
"pysbd",
|
||||||
"soundfile",
|
"soundfile",
|
||||||
"phonemizer>=2.2.0",
|
"phonemizer>=2.2.0",
|
||||||
"nose==1.3.7",
|
"nose==1.3.7",
|
||||||
|
|
|
@ -3,7 +3,6 @@
|
||||||
import argparse
|
import argparse
|
||||||
|
|
||||||
from TTS.utils.io import load_config
|
from TTS.utils.io import load_config
|
||||||
from TTS.utils.text.symbols import symbols, phonemes
|
|
||||||
from TTS.vocoder.tf.utils.generic_utils import setup_generator
|
from TTS.vocoder.tf.utils.generic_utils import setup_generator
|
||||||
from TTS.vocoder.tf.utils.io import load_checkpoint
|
from TTS.vocoder.tf.utils.io import load_checkpoint
|
||||||
from TTS.vocoder.tf.utils.tflite import convert_melgan_to_tflite
|
from TTS.vocoder.tf.utils.tflite import convert_melgan_to_tflite
|
||||||
|
|
|
@ -1,7 +1,6 @@
|
||||||
import tensorflow as tf
|
import tensorflow as tf
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
def convert_melgan_to_tflite(model,
|
def convert_melgan_to_tflite(model,
|
||||||
output_path=None,
|
output_path=None,
|
||||||
experimental_converter=True):
|
experimental_converter=True):
|
||||||
|
@ -25,6 +24,7 @@ def convert_melgan_to_tflite(model,
|
||||||
return None
|
return None
|
||||||
return tflite_model
|
return tflite_model
|
||||||
|
|
||||||
|
|
||||||
def load_tflite_model(tflite_path):
|
def load_tflite_model(tflite_path):
|
||||||
tflite_model = tf.lite.Interpreter(model_path=tflite_path)
|
tflite_model = tf.lite.Interpreter(model_path=tflite_path)
|
||||||
tflite_model.allocate_tensors()
|
tflite_model.allocate_tensors()
|
||||||
|
|
Loading…
Reference in New Issue