| 18 | |
| 19 | |
| 20 | def create_argparser(): |
| 21 | def convert_boolean(x): |
| 22 | return x.lower() in ["true", "1", "yes"] |
| 23 | |
| 24 | parser = argparse.ArgumentParser() |
| 25 | parser.add_argument( |
| 26 | "--list_models", |
| 27 | type=convert_boolean, |
| 28 | nargs="?", |
| 29 | const=True, |
| 30 | default=False, |
| 31 | help="list available pre-trained tts and vocoder models.", |
| 32 | ) |
| 33 | parser.add_argument( |
| 34 | "--model_name", |
| 35 | type=str, |
| 36 | default="tts_models/en/ljspeech/tacotron2-DDC", |
| 37 | help="Name of one of the pre-trained tts models in format <language>/<dataset>/<model_name>", |
| 38 | ) |
| 39 | parser.add_argument("--vocoder_name", type=str, default=None, help="name of one of the released vocoder models.") |
| 40 | |
| 41 | # Args for running custom models |
| 42 | parser.add_argument("--config_path", default=None, type=str, help="Path to model config file.") |
| 43 | parser.add_argument( |
| 44 | "--model_path", |
| 45 | type=str, |
| 46 | default=None, |
| 47 | help="Path to model file.", |
| 48 | ) |
| 49 | parser.add_argument( |
| 50 | "--vocoder_path", |
| 51 | type=str, |
| 52 | help="Path to vocoder model file. If it is not defined, model uses GL as vocoder. Please make sure that you installed vocoder library before (WaveRNN).", |
| 53 | default=None, |
| 54 | ) |
| 55 | parser.add_argument("--vocoder_config_path", type=str, help="Path to vocoder model config file.", default=None) |
| 56 | parser.add_argument("--speakers_file_path", type=str, help="JSON file for multi-speaker model.", default=None) |
| 57 | parser.add_argument("--port", type=int, default=5002, help="port to listen on.") |
| 58 | parser.add_argument("--use_cuda", type=convert_boolean, default=False, help="true to use CUDA.") |
| 59 | parser.add_argument("--debug", type=convert_boolean, default=False, help="true to enable Flask debug mode.") |
| 60 | parser.add_argument("--show_details", type=convert_boolean, default=False, help="Generate model detail page.") |
| 61 | return parser |
| 62 | |
| 63 | |
| 64 | # parse the args |