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

---
 .../.prompt_tuning.py.swp                     | Bin 86016 -> 86016 bytes
 .../prompt_tuning.py                          |   2 ++
 2 files changed, 2 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 1122ed489072eec1ebee6a10c5adf907024a9405..efbe37321ef1b9398ec67e78d9d52b26e0b943eb 100644
GIT binary patch
delta 567
zcmXBRO(?^09LMqR-|WFNV+YSWcrLU5m?94)2UDUQrkt=HP&R4HfttdBlT0Q_gg7ab
zT|AV7gOi9HL=J8q4i1*Xc-MaQ`t_;rw{Ly>PH$?bH?^igtIN@4Zm?K|mnmY?**$%S
zrhKCkm{E-p)u`MRrhQR+Nr*QD!dX_OI9QS)qJ<!uF7k#oETSD{xK9(A#S|vdh*+w~
zCC;#fVGN-PrO-o%#}ttbtYQQ%bfXh(DH0Sbp$Hm$CW}1b8W)J*1b(Q9CW(CG2#4^Z
z26{XgME0?VIn=|B9Ax8H&+fRwE@t7@2Ss`aDvbEiF$l+4$2`37z>OAULqSw4@{SuE
zV;K{0q8bJ~C$b23u!055!v`;VoA_!`fPCa3rV;tTEe^1a0rbNK2Vz|DCH$%?r=jyL
wL#Wtki~QM?s?^-Eao^~e-`s4D2aDfVQSJ_S2Fk;ZN~JE;;q^zZTb26Ef5%u}@Bjb+

delta 531
zcmXBPK`6rk7{=l6j~TWx)3V4+%A~BZQAV67YO%=<4i0M<4#;Ybo94g{lcsjyU^NF$
zNG53)L=FnYNyEXxNs1g~e&ws@?LEAgFS4nMY^scYyKm5K>9W~{(~H=OI|8kH##)0C
zjJ6pvZ3gA8PFJmn<O}`>!DOXfsSjH!L`+cOxLjlx0Sw?%Cz8en)-et<UbG^Q*vB%K
zFob?s(S&b}NESzkA%q!B!H*I2Y9uU;lydx(iM%0)2i#)~9+*&t+ftDm#4!vjzDh(c
zae*!PFa{?as6k#W@``h$5ydP#>aa*Vr5azwB6m2!HsV-B6cccu0VZhiu40b^Bryj!
zn(<r2BRIn$cCd{V#4(Ri^f-xbxL}43AB7^<IK=`Y2w@Ut<de^pifi>LmdK_`67^nh
TCfBaCXcM*5smxWM(y9Lg@gh}!

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 e991d5fc7..5b0917b84 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
@@ -1615,6 +1615,8 @@ def run_adapter(rank, world_size, args, wb=None):
     optimizer, scheduler = optimizer_adapter, scheduler_adapter
     
     librispeech = LibriSpeechAsrDataModule(args)
+
+    prompt = prompt.to('cuda')
     
     '''
     if params.hpo: