From 0917bd9d790f9df071fb9ce1e3b1f24ac324cc79 Mon Sep 17 00:00:00 2001
From: dohe0342 <kimdohe1070@gmail.com>
Date: Tue, 11 Apr 2023 13:47:46 +0900
Subject: [PATCH] from local

---
 .../.data2vec_encoder.py.swp                  | Bin 20480 -> 20480 bytes
 .../data2vec_encoder.py                       |   2 +-
 2 files changed, 1 insertion(+), 1 deletion(-)

diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.data2vec_encoder.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.data2vec_encoder.py.swp
index d6ac48fae66a21f53f91be055768e4edbb8dfadf..e62d6b83c82e8efe029543732ec3dc172e42227c 100644
GIT binary patch
delta 190
zcmZozz}T>WQ9Q{Y%+puFQqO<^2m}}yqMw<h{LY%XQT!)AquFFZ0eMD;&4vP(`FXiH
z85sTpF-VcwW<iC|^1K__7#Pm7LxdbA3p#A~jpJfqFalzEAm#w#J3tM$f%q&CF9PB^
zAPxj#2Ot&#Vm2Ut!NI^V4~XXiaT*Y(0x=g5pWQ6zxPxu;C->P*Ecv+wC6lLl$#3pb
LiDRAo(q9|^I}9mO

delta 178
zcmZozz}T>WQ9Q{Y%+puFQqO<^2m}}yT8`_bME;(*QT!)Aqxob(0eMFI&4vP(`FTHZ
zFfhE}geWrKEU55Vp7$0T1H&hFh>-ncL5J;L;am(1E<mgc#Ed}90K}hwnvMhUVjwO9
z;#43u2V!j?<^|%zK!XnfaW@cm0kJ+1e`4Rv$nt<~^I7-VOq+XDVpu1?^cM#J$f711

diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/data2vec_encoder.py b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/data2vec_encoder.py
index 107e06455..51f034fab 100644
--- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/data2vec_encoder.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/data2vec_encoder.py
@@ -52,7 +52,6 @@ class FairSeqData2VecEncoder(EncoderInterface):
         output_size: int = 256,
         freeze_finetune_updates: int = 0,
         additional_block: bool = False,
-        prompt = None,
     ):
         assert check_argument_types()
         super().__init__()
@@ -97,6 +96,7 @@ class FairSeqData2VecEncoder(EncoderInterface):
         x_lens: torch.Tensor,
         warmup = None,
         prev_states: torch.Tensor = None,
+        prompt: torch.Tensor = None,
     ) -> Tuple[torch.Tensor, torch.Tensor]:
         xs_pad = x
         ilens = x_lens