From ca29f9f9c4a50105ebe94d35fe1545a7aa68db76 Mon Sep 17 00:00:00 2001
From: dohe0342 <kimdohe1070@gmail.com>
Date: Tue, 11 Apr 2023 16:28:23 +0900
Subject: [PATCH] from local

---
 .../.prompt_tuning.py.swp                     | Bin 98304 -> 102400 bytes
 .../prompt_tuning.py                          |   6 ++++++
 2 files changed, 6 insertions(+)

diff --git 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
index 946cd7f676254df442e5e80a9f70b9df9d386a53..868970a9b0381ff5f57b11f3455fcf35008e7e91 100644
GIT binary patch
delta 874
zcmXxjT}V@59LMqhUOb&`&P`!X3geVFt#B>cf{MZnnq4eek+ce#W*O*eSp-IQPNqqE
z(Zzv|H$~-5NknTGK?Yet&`l5(WYNn_S2eInP*K_U6g}|aJTE-(JkS4k3_~$4E1}L<
z+t!q}HX0LJP~?*@(N!vhN&!dMbOlq6xS0-ytLyAl+OB^e*;vd3M*_Dsch<kpBhu<+
zp3QNy+IXZ}jLx7JJ7KS`LJVQdv!jI9xQ+>2#X02BjwEVO2_MR;$P6Ch2D;IKERtA>
zdU#aP<=q02S9p%QxPtTO$2PPei6}yFVO9~jh9hXfdL$6XA~-QyA@UXz*o_pD*n&pX
zL8-`!Javmqq8DorK?9b;gR)EH0}2?!Fow{BdW2DnUrv#ac!nwDvCdgsJ-ASHX_{uH
zkJp+f2O>(o?{F@E%uJtHQM^C8>8sw+B~rzKN4c0?m;Ys(4~YE6XFSFTPQySqTCoAk
z@x4l94rR<>8aH8~6ML}-&4{1|Km8(i{8XSo=N2sJ$RUF$mZ1S&e5n+9k0-c<aacHo
z{n&vfeDjGsz&OS*j3M-3JDSjlIpQ_x)kQASv2%C<12~8_EXEJ=bPs3IkE7^A2F-}U
sfj8voC0<|(IoP>7fD{r~0w1P%>A$aS{~$%x?N=<L-JupY4<1tg0EF&=ZvX%Q

delta 442
zcmXBQKP&?Q9L4e9wWn7{(-Y|*Ax)E-3{|53PZoDWiN#`2jlsZB3$gSh9BB;XW{J=R
z5z!6?5<^#sL@Y@BS!^UtV-;WVlF#;rmuDaNY}*&grKYV+d?=X`Vu;){rRT>kLZ3k;
z?0Sui;<!^rq_S<S2KV0L<+z#+XcR|0|82)@v8#o1U1T3C7(_365QX5YQREpXDB%c2
z<S`2qtq7nQb-&0HsyM>}a+rgO6k^cvs!2f}+4O)b9AX!1n1lrr-3UWP%_nk<Wmp(R
zKho%c7qteF7nCrA378m15^d0Mr)EV?RgonOpbMQyKu6sx@{Tfg;9wJrh#`sye)#kQ
a_qawLe?Jd<F4wn|g5OPSv{!D6ec!*8K0-YJ

diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/prompt_tuning.py b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/prompt_tuning.py
index 39820a8aa..3b4b678a6 100755
--- 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(