🤾 Update Dockerfile, setup.py to TensorFlow 2.6.0

Author: dathudeptrai

README.md

@@ -60,10 +60,10 @@ This repository is tested on Ubuntu 18.04 with:
 - Python 3.7+
 - Cuda 10.1
 - CuDNN 7.6.5
-- Tensorflow 2.2/2.3
+- Tensorflow 2.2/2.3/2.4/2.5/2.6
 - [Tensorflow Addons](https://github.com/tensorflow/addons) >= 0.10.0
 
-Different Tensorflow version should be working but not tested yet. This repo will try to work with the latest stable TensorFlow version. **We recommend you install TensorFlow 2.3.0 to training in case you want to use MultiGPU.**
+Different Tensorflow version should be working but not tested yet. This repo will try to work with the latest stable TensorFlow version. **We recommend you install TensorFlow 2.6.0 to training in case you want to use MultiGPU.**
 
 ## Installation
 ### With pip

docker-compose.yml

@@ -1,4 +1,4 @@
-version: '2.3'
+version: '2.6'
 services:
   tensorflowtts:
     build: .

dockerfile

@@ -1,7 +1,8 @@
-FROM tensorflow/tensorflow:2.3.1-gpu
+FROM tensorflow/tensorflow:2.6.0-gpu
 RUN apt-get update
 RUN apt-get install -y zsh tmux wget git libsndfile1
 RUN pip install ipython && \
-    pip install git+https://github.com/TensorSpeech/TensorflowTTS.git
+    pip install git+https://github.com/TensorSpeech/TensorflowTTS.git && \
+    pip install git+https://github.com/repodiac/german_transliterate.git#egg=german_transliterate
 RUN mkdir /workspace
 WORKDIR /workspace

setup.py

@@ -9,7 +9,7 @@
 
 if LooseVersion(sys.version) < LooseVersion("3.6"):
     raise RuntimeError(
-        "Tensorflow TTS requires python >= 3.6, "
+        "TensorFlow TTS requires python >= 3.6, "
         "but your Python version is {}".format(sys.version)
     )
 
@@ -22,7 +22,7 @@
 # TODO(@dathudeptrai) update requirement if needed.
 requirements = {
     "install": [
-        "tensorflow-gpu==2.3.1",
+        "tensorflow-gpu==2.6.0",
         "tensorflow-addons>=0.10.0",
         "setuptools>=38.5.1",
         "huggingface_hub==0.0.8",
@@ -73,9 +73,9 @@
     name="TensorFlowTTS",
     version="0.0",
     url="https://github.com/tensorspeech/TensorFlowTTS",
-    author="Minh Nguyen Quan Anh, Eren Gölge, Kuan Chen, Dawid Kobus, Takuya Ebata, Trinh Le Quang, Yunchao He, Alejandro Miguel Velasquez",
+    author="Minh Nguyen Quan Anh, Alejandro Miguel Velasquez, Dawid Kobus, Eren Gölge, Kuan Chen, Takuya Ebata, Trinh Le Quang, Yunchao He",
     author_email="nguyenquananhminh@gmail.com",
-    description="TensorFlowTTS: Real-Time State-of-the-art Speech Synthesis for Tensorflow 2",
+    description="TensorFlowTTS: Real-Time State-of-the-art Speech Synthesis for TensorFlow 2",
     long_description=open(os.path.join(dirname, "README.md"), encoding="utf-8").read(),
     long_description_content_type="text/markdown",
     license="Apache-2.0",
@@ -88,6 +88,7 @@
     classifiers=[
         "Programming Language :: Python :: 3.6",
         "Programming Language :: Python :: 3.7",
+        "Programming Language :: Python :: 3.8",
         "Intended Audience :: Science/Research",
         "Operating System :: POSIX :: Linux",
         "License :: OSI Approved :: Apache Software License",