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

---
 .../.prompt_tuning.py.swp                     | Bin 90112 -> 94208 bytes
 .../prompt_tuning.py                          |   5 +----
 2 files changed, 1 insertion(+), 4 deletions(-)

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 4fc3de0771bf7202118acf592888803f8aefb177..2b3fb91a844d8c8a717aba83895212f5965cf7ca 100644
GIT binary patch
delta 902
zcmXxiUr1A77{~GF?M$7`HMd*}sf<EKO=Jz4{b4AJ{%DkxI?-TIX+)OET0}-{SJ_1x
zxoVLYN?o)r4D6J<5tFW3b!RKei*!O{VE>4q5!H7{4}8vf5AWeU56|yJr}Lt7dG;gj
zK*K)A)*81^#Ugd4&7Jk*_9ct5>ss-WQo-0aYpHEbZksA!zDY%XENcy@kbb~+)%?(5
zaG0A6A`5wJb4XCP)?ZMO+~X9kqYVL6;s>v0@ECnKj~!TzSxqF3S9pR1E@3;os6i<N
z=@laVIEp$LkSoV>=@!4P#lG&Wg40~>=nb7d)f@J;gwA%w#=RxFYuDRF^EO3Kx$|f{
z6t}w7`bfD&q!3f9L>}Qd8u80aI~c_Xf>?(SMI!T<zyPjbKN{e|M*KF3JjEmi(T6kW
z!7;SJ52r~YVq+orypoY(7VmHittdw+UKcV0451k=WQ`&(@dEeJfp%;`Eh?emQ-Med
z3G^e3R(J~{B5PS>^F`iZ8e_PNJGh90s7Ebau;4qv&R_!9a1y&wfj<PhfHYEgh6fnK
zO@wd=9cXuR@u3=4ERjwIqZq*rT!s(1<h11cCRL^48ILMUzW1nMO~2RGZH({TpggfX
zJ>lf@I^}gl4tJ6aLoFkXT-g8r<i8}HKsQdn10#NtcnX6UKpUL+LgI_~h#5@cHk!}~
hKU^rmBHtRuxgsf$W)@{dx-txz;?bZo>c##!?H{u4lz0FD

delta 749
zcmXBRT}YEr7{Kx8^lomp*4$J^uHsPmQJ}q64ictOK`w~Iiv<ahZqWC(Vk6LMUf6|p
z(Z?Gp5#3bOMFjdnFJ6RE(1mmnL8TD7F7m2n(M6*R>2DtR9iGE^&U4P;87jJmUb$=g
z%|!gL-qT_Vqe>*^YfFVEYD)oCV;fZ^r3S2D!LZd7>QZ_8Lg<G7R#8uCm-f5J_pVtz
zG++n!4ywXWx5yI4(Tii)f-<>_n8H<D!9j$vV2I3N7H{wjqv%9Cb|L~7=D6-UPGUd)
zU%&EhhSu5&^t&B>P8AI&Tc=`%-5meC=wD)Uf#j9X<;Q}i+FEc2SfadM<QwkcCJvwh
zAN?Z3xQs!xV|ks(YrMp5oW?0^M?E~Kz^qT?0mg9#2^_&8?1Sl(f@~%*@MEpW7tCN9
z4qhOKR+KA6eq$07NMi>)_~aFNiYK^+DB4hiYW(%^Z@k503?qqdPl4hDA^fSJ5NzDX
z2+~L)i4Iglp-d;g<2`KL!6o#d8D7lO(P=!wU5sJ`8KiL(yG*8yh@cM3w6=r~c#enY
z!wDo1cfPi&dS!R?<P7WGe8icGs<U2e)A7m;*^8O;*_@8))u1JpX=+U7Q+<tN@oj3i
N)s@aUGo31`{{e~2f3g4o

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 d5918632a..afaec6557 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
@@ -1560,7 +1560,7 @@ def run_adapter(rank, world_size, args, wb=None):
     logging.info(params)
 
     logging.info("About to create model")
-    model = get_transducer_model(params)
+    model = get_transducer_model(params, prompt=True)
 
     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}")
@@ -1618,8 +1618,6 @@ def run_adapter(rank, world_size, args, wb=None):
     
     librispeech = LibriSpeechAsrDataModule(args)
 
-    prompt = prompt.to('cuda')
-    
     '''
     if params.hpo:
         train_cuts = librispeech.train_clean_10_cuts(option=params.gender)
@@ -1689,7 +1687,6 @@ def run_adapter(rank, world_size, args, wb=None):
             world_size=world_size,
             rank=rank,
             wb=wb,
-            prompt=prompt,
         )
 
         if params.print_diagnostics: