From 1bb0bf7e8bb05a7cb7270081561cee29880277b3 Mon Sep 17 00:00:00 2001 From: baojia tong Date: Fri, 23 Jun 2023 11:39:37 -0400 Subject: [PATCH] fix cli optional type check --- bark/cli.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/bark/cli.py b/bark/cli.py index 687e565..924af22 100644 --- a/bark/cli.py +++ b/bark/cli.py @@ -20,9 +20,9 @@ def cli(): parser.add_argument("--output_dir", type=str, default=".", help="directory to save the outputs") parser.add_argument( "--history_prompt", - type=Optional[Union[Dict, str]], + type=str, default=None, - help="history choice for audio cloning", + help="history choice for audio cloning, be path to the .npz file.", ) parser.add_argument( "--text_temp", @@ -48,7 +48,7 @@ def cli(): input_text: str = args.get("text") output_filename: str = args.get("output_filename") output_dir: str = args.get("output_dir") - history_prompt: Optional[Union[Dict, str]] = args.get("history_prompt") + history_prompt: str = args.get("history_prompt") text_temp: float = args.get("text_temp") waveform_temp: float = args.get("waveform_temp") silent: bool = args.get("silent")