From bf0ec25fa8b2b03ef72dc7ddf661197bcb3c2684 Mon Sep 17 00:00:00 2001 From: "github-actions[bot]" <41898282+github-actions[bot]@users.noreply.github.com> Date: Fri, 14 Feb 2025 00:14:34 +0900 Subject: [PATCH] chore(format): run black on dev (#894) Co-authored-by: github-actions[bot] --- examples/api/postScript.py | 189 ++++++++++++++++++++++++++++++------- examples/web/webui.py | 4 +- 2 files changed, 156 insertions(+), 37 deletions(-) diff --git a/examples/api/postScript.py b/examples/api/postScript.py index 2f12f3926..cfc2768b7 100644 --- a/examples/api/postScript.py +++ b/examples/api/postScript.py @@ -14,49 +14,168 @@ def parse_arguments(): parser = argparse.ArgumentParser(description="HTTP client for ChatTTS service") - parser.add_argument("--text", type=str, nargs='+', required=True, help="Text to synthesize") - parser.add_argument("--audio_seed", type=int, required=True, help="Audio generation seed") - parser.add_argument("--text_seed", type=int, required=True, help="Text generation seed") - parser.add_argument("--stream", type=bool, default=False, help="Enable/disable streaming") + parser.add_argument( + "--text", type=str, nargs="+", required=True, help="Text to synthesize" + ) + parser.add_argument( + "--audio_seed", type=int, required=True, help="Audio generation seed" + ) + parser.add_argument( + "--text_seed", type=int, required=True, help="Text generation seed" + ) + parser.add_argument( + "--stream", type=bool, default=False, help="Enable/disable streaming" + ) parser.add_argument("--lang", type=str, default=None, help="Language code for text") - parser.add_argument("--skip_refine_text", type=bool, default=True, help="Skip text refinement") - parser.add_argument("--refine_text_only", type=bool, default=False, help="Only refine text") - parser.add_argument("--use_decoder", type=bool, default=True, help="Use decoder during inference") - parser.add_argument("--do_text_normalization", type=bool, default=True, help="Enable text normalization") - parser.add_argument("--do_homophone_replacement", type=bool, default=False, help="Enable homophone replacement") - parser.add_argument("--tgt", type=str, default="./output", help="Target directory to save output files") - parser.add_argument("--filename", type=str, default="test.mp3", help="Target directory to save output files") + parser.add_argument( + "--skip_refine_text", type=bool, default=True, help="Skip text refinement" + ) + parser.add_argument( + "--refine_text_only", type=bool, default=False, help="Only refine text" + ) + parser.add_argument( + "--use_decoder", type=bool, default=True, help="Use decoder during inference" + ) + parser.add_argument( + "--do_text_normalization", + type=bool, + default=True, + help="Enable text normalization", + ) + parser.add_argument( + "--do_homophone_replacement", + type=bool, + default=False, + help="Enable homophone replacement", + ) + parser.add_argument( + "--tgt", + type=str, + default="./output", + help="Target directory to save output files", + ) + parser.add_argument( + "--filename", + type=str, + default="test.mp3", + help="Target directory to save output files", + ) # Refinement text parameters - parser.add_argument("--refine_prompt", type=str, default="", help="Prompt for text refinement") - parser.add_argument("--refine_top_P", type=float, default=0.7, help="Top P value for text refinement") - parser.add_argument("--refine_top_K", type=int, default=20, help="Top K value for text refinement") - parser.add_argument("--refine_temperature", type=float, default=0.7, help="Temperature for text refinement") - parser.add_argument("--refine_repetition_penalty", type=float, default=1.0, - help="Repetition penalty for text refinement") - parser.add_argument("--refine_max_new_token", type=int, default=384, help="Max new tokens for text refinement") - parser.add_argument("--refine_min_new_token", type=int, default=0, help="Min new tokens for text refinement") - parser.add_argument("--refine_show_tqdm", type=bool, default=True, help="Show progress bar for text refinement") - parser.add_argument("--refine_ensure_non_empty", type=bool, default=True, help="Ensure non-empty output") - parser.add_argument("--refine_stream_batch", type=int, default=24, help="Stream batch size for refinement") + parser.add_argument( + "--refine_prompt", type=str, default="", help="Prompt for text refinement" + ) + parser.add_argument( + "--refine_top_P", + type=float, + default=0.7, + help="Top P value for text refinement", + ) + parser.add_argument( + "--refine_top_K", type=int, default=20, help="Top K value for text refinement" + ) + parser.add_argument( + "--refine_temperature", + type=float, + default=0.7, + help="Temperature for text refinement", + ) + parser.add_argument( + "--refine_repetition_penalty", + type=float, + default=1.0, + help="Repetition penalty for text refinement", + ) + parser.add_argument( + "--refine_max_new_token", + type=int, + default=384, + help="Max new tokens for text refinement", + ) + parser.add_argument( + "--refine_min_new_token", + type=int, + default=0, + help="Min new tokens for text refinement", + ) + parser.add_argument( + "--refine_show_tqdm", + type=bool, + default=True, + help="Show progress bar for text refinement", + ) + parser.add_argument( + "--refine_ensure_non_empty", + type=bool, + default=True, + help="Ensure non-empty output", + ) + parser.add_argument( + "--refine_stream_batch", + type=int, + default=24, + help="Stream batch size for refinement", + ) # Infer code parameters - parser.add_argument("--infer_prompt", type=str, default="[speed_5]", help="Prompt for inference") - parser.add_argument("--infer_top_P", type=float, default=0.1, help="Top P value for inference") - parser.add_argument("--infer_top_K", type=int, default=20, help="Top K value for inference") - parser.add_argument("--infer_temperature", type=float, default=0.3, help="Temperature for inference") - parser.add_argument("--infer_repetition_penalty", type=float, default=1.05, help="Repetition penalty for inference") - parser.add_argument("--infer_max_new_token", type=int, default=2048, help="Max new tokens for inference") - parser.add_argument("--infer_min_new_token", type=int, default=0, help="Min new tokens for inference") - parser.add_argument("--infer_show_tqdm", type=bool, default=True, help="Show progress bar for inference") - parser.add_argument("--infer_ensure_non_empty", type=bool, default=True, help="Ensure non-empty output") - parser.add_argument("--infer_stream_batch", type=bool, default=True, help="Stream batch for inference") - parser.add_argument("--infer_spk_emb", type=str, default=None, help="Speaker embedding for inference") + parser.add_argument( + "--infer_prompt", type=str, default="[speed_5]", help="Prompt for inference" + ) + parser.add_argument( + "--infer_top_P", type=float, default=0.1, help="Top P value for inference" + ) + parser.add_argument( + "--infer_top_K", type=int, default=20, help="Top K value for inference" + ) + parser.add_argument( + "--infer_temperature", type=float, default=0.3, help="Temperature for inference" + ) + parser.add_argument( + "--infer_repetition_penalty", + type=float, + default=1.05, + help="Repetition penalty for inference", + ) + parser.add_argument( + "--infer_max_new_token", + type=int, + default=2048, + help="Max new tokens for inference", + ) + parser.add_argument( + "--infer_min_new_token", + type=int, + default=0, + help="Min new tokens for inference", + ) + parser.add_argument( + "--infer_show_tqdm", + type=bool, + default=True, + help="Show progress bar for inference", + ) + parser.add_argument( + "--infer_ensure_non_empty", + type=bool, + default=True, + help="Ensure non-empty output", + ) + parser.add_argument( + "--infer_stream_batch", + type=bool, + default=True, + help="Stream batch for inference", + ) + parser.add_argument( + "--infer_spk_emb", + type=str, + default=None, + help="Speaker embedding for inference", + ) return parser.parse_args() - def main(): args = parse_arguments() @@ -65,7 +184,7 @@ def main(): "text": args.text, "stream": args.stream, "lang": args.lang, - "filename":args.filename, + "filename": args.filename, "skip_refine_text": args.skip_refine_text, "refine_text_only": args.refine_text_only, "use_decoder": args.use_decoder, diff --git a/examples/web/webui.py b/examples/web/webui.py index a2864adb0..a2bcf9e67 100644 --- a/examples/web/webui.py +++ b/examples/web/webui.py @@ -102,7 +102,7 @@ def main(): minimum=seed_min, maximum=seed_max, ) - generate_audio_seed = gr.Button("\U0001F3B2", interactive=True) + generate_audio_seed = gr.Button("\U0001f3b2", interactive=True) text_seed_input = gr.Number( value=ex[0][5], label="Text Seed", @@ -110,7 +110,7 @@ def main(): minimum=seed_min, maximum=seed_max, ) - generate_text_seed = gr.Button("\U0001F3B2", interactive=True) + generate_text_seed = gr.Button("\U0001f3b2", interactive=True) with gr.Row(): spk_emb_text = gr.Textbox(