pylint fixes and add missing requirement

This commit is contained in:
erogol 2020-07-14 17:55:46 +02:00
parent b2cc256dab
commit 664f42df33
5 changed files with 5 additions and 5 deletions

View File

@ -15,6 +15,7 @@ tqdm
inflect
pysbd
bokeh==1.4.0
pysbd
soundfile
nose==1.3.7
cardboardlint==1.3.0

View File

@ -1,5 +1,4 @@
import io
import re
import sys
import time

View File

@ -93,6 +93,7 @@ requirements = {
"inflect",
"pysbd",
"bokeh==1.4.0",
"pysbd",
"soundfile",
"phonemizer>=2.2.0",
"nose==1.3.7",

View File

@ -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

View File

@ -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()