mirror of
https://github.com/wassname/TTS.git
synced 2026-09-09 11:16:00 +08:00
Remove Tensorflow requeriment (#1225)
* Remove TF modules * Remove TF unit tests * Remove TF vocoder modules * Remove TF convert scripts * Remove TF requirement * Remove the Docs TF instructions * Remove TF inference support
This commit is contained in:
@@ -1,21 +0,0 @@
|
||||
# Converting Torch to TF 2
|
||||
|
||||
Currently, 🐸TTS supports the vanilla Tacotron2 and MelGAN models in TF 2.It does not support advanced attention methods and other small tricks used by the Torch models. You can convert any Torch model trained after v0.0.2.
|
||||
|
||||
You can also export TF 2 models to TFLite for even faster inference.
|
||||
|
||||
## How to convert from Torch to TF 2.0
|
||||
Make sure you installed Tensorflow v2.2. It is not installed by default by :frog: TTS.
|
||||
|
||||
All the TF related code stays under ```tf``` folder.
|
||||
|
||||
To convert a **compatible** Torch model, run the following command with the right arguments:
|
||||
|
||||
```bash
|
||||
python TTS/bin/convert_tacotron2_torch_to_tf.py\
|
||||
--torch_model_path /path/to/torch/model.pth.tar \
|
||||
--config_path /path/to/model/config.json\
|
||||
--output_path /path/to/output/tf/model
|
||||
```
|
||||
|
||||
This will create a TF model file. Notice that our model format is not compatible with the official TF checkpoints. We created our custom format to match Torch checkpoints we use. Therefore, use the ```load_checkpoint``` and ```save_checkpoint``` functions provided under ```TTS.tf.generic_utils```.
|
||||
@@ -27,7 +27,6 @@
|
||||
formatting_your_dataset
|
||||
what_makes_a_good_dataset
|
||||
tts_datasets
|
||||
converting_torch_to_tf
|
||||
|
||||
.. toctree::
|
||||
:maxdepth: 2
|
||||
|
||||
@@ -12,12 +12,6 @@ You can install from PyPI as follows:
|
||||
pip install TTS # from PyPI
|
||||
```
|
||||
|
||||
By default, this only installs the requirements for PyTorch. To install the tensorflow dependencies as well, use the `tf` extra.
|
||||
|
||||
```bash
|
||||
pip install TTS[tf]
|
||||
```
|
||||
|
||||
Or install from Github:
|
||||
|
||||
```bash
|
||||
|
||||
Reference in New Issue
Block a user