Skip to content

Commit

Permalink
chore(format): run black on dev (#894)
Browse files Browse the repository at this point in the history 
Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
  • Loading branch information
@github-actions
github-actions[bot] and github-actions[bot] authored Feb 13, 2025
1 parent 6bacab1 commit bf0ec25
Show file tree
Hide file tree
Showing 2 changed files with 156 additions and 37 deletions.
189 changes: 154 additions & 35 deletions examples/api/postScript.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -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()

Expand All @@ -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,
Expand Down
4 changes: 2 additions & 2 deletions examples/web/webui.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -102,15 +102,15 @@ 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",
interactive=True,
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(
Expand Down

0 comments on commit bf0ec25

Please sign in to comment.