From 76ed34ddd2e610d466cf2c6fb76e13eea6df27bb Mon Sep 17 00:00:00 2001 From: mrq Date: Sun, 11 Jun 2023 04:46:22 +0000 Subject: [PATCH] added CLI script (python ./src/cli.py --text=TEXT --voice=VOICE' etc) --- src/cli.py | 66 ++++++++++++++++++++++++++++++++++++++++++++++++++++ src/utils.py | 9 ++++--- 2 files changed, 72 insertions(+), 3 deletions(-) create mode 100755 src/cli.py diff --git a/src/cli.py b/src/cli.py new file mode 100755 index 0000000..d463edc --- /dev/null +++ b/src/cli.py @@ -0,0 +1,66 @@ +import os +import argparse + +if 'TORTOISE_MODELS_DIR' not in os.environ: + os.environ['TORTOISE_MODELS_DIR'] = os.path.realpath(os.path.join(os.getcwd(), './models/tortoise/')) + +if 'TRANSFORMERS_CACHE' not in os.environ: + os.environ['TRANSFORMERS_CACHE'] = os.path.realpath(os.path.join(os.getcwd(), './models/transformers/')) + +os.environ['PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION'] = 'python' + +from utils import * + +if __name__ == "__main__": + args = setup_args(cli=True) + + default_arguments = import_generate_settings() + parser = argparse.ArgumentParser(allow_abbrev=False) + parser.add_argument("--text", default=default_arguments['text']) + parser.add_argument("--delimiter", default=default_arguments['delimiter']) + parser.add_argument("--emotion", default=default_arguments['emotion']) + parser.add_argument("--prompt", default=default_arguments['prompt']) + parser.add_argument("--voice", default=default_arguments['voice']) + parser.add_argument("--mic_audio", default=default_arguments['mic_audio']) + parser.add_argument("--voice_latents_chunks", default=default_arguments['voice_latents_chunks']) + parser.add_argument("--candidates", default=default_arguments['candidates']) + parser.add_argument("--seed", default=default_arguments['seed']) + parser.add_argument("--num_autoregressive_samples", default=default_arguments['num_autoregressive_samples']) + parser.add_argument("--diffusion_iterations", default=default_arguments['diffusion_iterations']) + parser.add_argument("--temperature", default=default_arguments['temperature']) + parser.add_argument("--diffusion_sampler", default=default_arguments['diffusion_sampler']) + parser.add_argument("--breathing_room", default=default_arguments['breathing_room']) + parser.add_argument("--cvvp_weight", default=default_arguments['cvvp_weight']) + parser.add_argument("--top_p", default=default_arguments['top_p']) + parser.add_argument("--diffusion_temperature", default=default_arguments['diffusion_temperature']) + parser.add_argument("--length_penalty", default=default_arguments['length_penalty']) + parser.add_argument("--repetition_penalty", default=default_arguments['repetition_penalty']) + parser.add_argument("--cond_free_k", default=default_arguments['cond_free_k']) + + args, unknown = parser.parse_known_args() + kwargs = { + 'text': args.text, + 'delimiter': args.delimiter, + 'emotion': args.emotion, + 'prompt': args.prompt, + 'voice': args.voice, + 'mic_audio': args.mic_audio, + 'voice_latents_chunks': args.voice_latents_chunks, + 'candidates': args.candidates, + 'seed': args.seed, + 'num_autoregressive_samples': args.num_autoregressive_samples, + 'diffusion_iterations': args.diffusion_iterations, + 'temperature': args.temperature, + 'diffusion_sampler': args.diffusion_sampler, + 'breathing_room': args.breathing_room, + 'cvvp_weight': args.cvvp_weight, + 'top_p': args.top_p, + 'diffusion_temperature': args.diffusion_temperature, + 'length_penalty': args.length_penalty, + 'repetition_penalty': args.repetition_penalty, + 'cond_free_k': args.cond_free_k, + 'experimentals': default_arguments['experimentals'], + } + + tts = load_tts() + generate(**kwargs) \ No newline at end of file diff --git a/src/utils.py b/src/utils.py index a204011..ac2209a 100755 --- a/src/utils.py +++ b/src/utils.py @@ -3008,7 +3008,7 @@ def get_args(): global args return args -def setup_args(): +def setup_args(cli=False): global args default_arguments = { @@ -3066,7 +3066,7 @@ def setup_args(): print(e) pass - parser = argparse.ArgumentParser() + parser = argparse.ArgumentParser(allow_abbrev=not cli) parser.add_argument("--share", action='store_true', default=default_arguments['share'], help="Lets Gradio return a public URL to use anywhere") parser.add_argument("--listen", default=default_arguments['listen'], help="Path for Gradio to listen on") parser.add_argument("--check-for-updates", action='store_true', default=default_arguments['check-for-updates'], help="Checks for update on startup") @@ -3108,7 +3108,10 @@ def setup_args(): parser.add_argument("--training-default-bnb", action='store_true', default=default_arguments['training-default-bnb'], help="Training default: bnb") parser.add_argument("--os", default="unix", help="Specifies which OS, easily") - args = parser.parse_args() + if cli: + args, unknown = parser.parse_known_args() + else: + args = parser.parse_args() args.embed_output_metadata = not args.no_embed_output_metadata