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

---
 .../.prompt_tuning.py.swp                     | Bin 102400 -> 102400 bytes
 .../prompt_tuning.py                          |   2 +-
 2 files changed, 1 insertion(+), 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 5d88b64b389eb87d122cff0f0d297dc66c29e082..faca55af5a0c264cd3fd02618f8b7552197e534d 100644
GIT binary patch
delta 57
zcmZozz}B#UO*F|M%+puFQqO<^2n2wL*ED5Eqv%#qMs|C)l+?1!<kab&nv5osJ@QJn
KpR#8(3jhF{^$&Ca

delta 57
zcmZozz}B#UO*F|M%+puFQqO<^2m}}yp7NNccs7b|6=h_%XH!ouO-WRr?y1RWGT9@q
LWcw+5Mza6_p@a`q

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 cb767546d..a94a95a13 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
@@ -1622,7 +1622,7 @@ def run_adapter(rank, world_size, args, wb=None):
     
     librispeech = LibriSpeechAsrDataModule(args)
 
-    model.prompt = model.prompt.to('cuda')
+    model.prompt = model.prompt.to(device)
 
     '''
     if params.hpo: