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

---
 .../.prompt_tuning.py.swp                     | Bin 94208 -> 98304 bytes
 .../prompt_tuning.py                          |   3 ++-
 2 files changed, 2 insertions(+), 1 deletion(-)

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 9b9b096ffe965dba6c3ad1142563f55c60b87875..946cd7f676254df442e5e80a9f70b9df9d386a53 100644
GIT binary patch
delta 775
zcmZwFT}V@L7{~GF@6K_~X?sa#Nx=%SpgLJ3xlGE|jacbLM3@AN5mD)e>jkfN#7o^k
zXDlM9tH4F%{KJMsgh@sb(Tfn|MRt{e!7l11i24q_?12yGIWPaid4A__FPpYq8Mx4V
zrt@SlWHk%r6?x`vi5|=41FIet$T+-@l$FVQ>wWd*+f;b%AoujFcZX*tXavnZL!_i#
zM@N(@=~PFrkvV<BUmR*s2ON4&=ckLcZ>mJ@yI9EUt&DH_NwR!4iZoIf!)0_MgiZL$
zs%eZPh5<C;i&NwlZsG=Z;(MjYbIhTLDGcE%dT;~^B~#?3DM=~Nq+p{1AvE9}vpmKn
zbYUZ0_~j7!i6RP^fQ>6Sg?*^Q=L(TUe8d8BNF#x>Xve_{a%iLx{8b{$Dm$)JUADDJ
zd30bV?c5(+ulQiBKYpz}9vO`EC;ofUT6<)-emWa2BqLEinLOIashoCr{crvDkP%k!
z4KI+xBoa6bH&!=@yv8dG;}A-<A|LP`BRGz&*aANcEY^@~4Y@v~xr07*VLN<Saf>YB
zE+%muyHJh!YLN#RKomXbh95rsU^6UX0aGY%mKa*F7Z$d`bTu+1lfQ+=ZA>7EbLd4|
Sl~%@PXVyNi4E-lR=llnr&4%&-

delta 416
zcmXBQO)EqJ6vpv$XXXx*#$4sS<i^rVBT*I_78E6#m}DVogvI0%W#<a9ATM_zAHYmZ
zWZXn%g$*mLgoP;^YG%a(D}OoltFt*zoyRoYW=bs__J=yU^rl+BFq+6vPD41dmncm8
zlmg4C9VmvC(2DFgt=PAj(vNkQu6DabUTsgyf}*_IYpnw=>$9!WTdf;Y6A@#&xp!)2
z#7Z>g+g0t4<K8l2zm@-N!^%5K&YgKgVu)fF3m8KXb@<5^xxqeGu#5n{vP2HCfpxUP
zfoHeK6Yg+{Ei7Ri{V0V8DOKcFl{tyB>|g*v)Z?9>t}%}=s!@SlcyZ_C7;)@i5u@lt
z4L%$q8Km)oE5xvhNrcdYW(OUYuoUC3moH`WzI@vpR9vTlS>+<jdPrvM_<7&SFQIr=
AI{*Lx

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 472a284b4..39820a8aa 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
@@ -666,6 +666,7 @@ def get_transducer_model(params: AttributeDict) -> nn.Module:
         decoder_dim=params.decoder_dim,
         joiner_dim=params.joiner_dim,
         vocab_size=params.vocab_size,
+        prompt=params.prompt,
     )
     return model
 
@@ -1558,7 +1559,7 @@ def run_adapter(rank, world_size, args, wb=None):
     logging.info(params)
 
     logging.info("About to create model")
-    model = get_transducer_model(params, prompt=True)
+    model = get_transducer_model(params)
 
     num_param = sum([p.numel() if p.requires_grad else 0 for p in model.parameters()])
     logging.info(f"Number of model parameters: {num_param}")