from local

2023-04-11 16:28:23 +09:00 · 2023-04-11 16:28:23 +09:00 · ca29f9f9c4
commit ca29f9f9c4
parent e1c696e134
2 changed files with 6 additions and 0 deletions
--- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.prompt_tuning.py.swp
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.prompt_tuning.py.swp
--- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/prompt_tuning.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/prompt_tuning.py
@ -314,6 +314,12 @@ def add_model_arguments(parser: argparse.ArgumentParser):
        """,
    )

+    parser.add_argument(
+        "--prompt",
+        type=str2bool,
+        default=False,
+    )
+

 def get_parser():
    parser = argparse.ArgumentParser(