mirror of
https://github.com/2noise/ChatTTS.git
synced 2026-08-30 17:05:27 +08:00
chore(format): run black on dev (#894)
Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
This commit is contained in:
github-actions[bot]
committed by
GitHub
GitHub
parent
6bacab1778
commit
bf0ec25fa8
+154
-35
@@ -14,49 +14,168 @@ CHATTTS_URL = f"http://{chattts_service_host}:{chattts_service_port}/generate_vo
|
||||
|
||||
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,
|
||||
|
||||
@@ -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(
|
||||
|
||||
Reference in New Issue
Block a user