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
|
||||
pysbd
|
||||
bokeh==1.4.0
|
||||
pysbd
|
||||
soundfile
|
||||
nose==1.3.7
|
||||
cardboardlint==1.3.0
|
||||
|
|
|
@ -1,5 +1,4 @@
|
|||
import io
|
||||
import re
|
||||
import sys
|
||||
import time
|
||||
|
||||
|
|
1
setup.py
1
setup.py
|
@ -93,6 +93,7 @@ requirements = {
|
|||
"inflect",
|
||||
"pysbd",
|
||||
"bokeh==1.4.0",
|
||||
"pysbd",
|
||||
"soundfile",
|
||||
"phonemizer>=2.2.0",
|
||||
"nose==1.3.7",
|
||||
|
|
|
@ -3,7 +3,6 @@
|
|||
import argparse
|
||||
|
||||
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.io import load_checkpoint
|
||||
from TTS.vocoder.tf.utils.tflite import convert_melgan_to_tflite
|
||||
|
|
|
@ -1,10 +1,9 @@
|
|||
import tensorflow as tf
|
||||
|
||||
|
||||
|
||||
def convert_melgan_to_tflite(model,
|
||||
output_path=None,
|
||||
experimental_converter=True):
|
||||
output_path=None,
|
||||
experimental_converter=True):
|
||||
"""Convert Tensorflow MelGAN model to TFLite. Save a binary file if output_path is
|
||||
provided, else return TFLite model."""
|
||||
|
||||
|
@ -25,6 +24,7 @@ def convert_melgan_to_tflite(model,
|
|||
return None
|
||||
return tflite_model
|
||||
|
||||
|
||||
def load_tflite_model(tflite_path):
|
||||
tflite_model = tf.lite.Interpreter(model_path=tflite_path)
|
||||
tflite_model.allocate_tensors()
|
||||
|
|
Loading…
Reference in New Issue