From 2e61476b01412f2c2969150f2059f63f737f0618 Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Tue, 11 Apr 2023 16:26:24 +0900 Subject: [PATCH] from local --- .../.model.py.swp | Bin 28672 -> 28672 bytes .../model.py | 3 +-- 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.model.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.model.py.swp index d9080af9879f2d6fc740a01797a38fe07acd5e21..6ea46cdcb81d00ad88e125531f6031fb9f46c633 100644 GIT binary patch delta 94 zcmZp8z}WDBQ8dXQ%+puFQqO<^2m}}yCUBdkyx1uEo1Zayv!K9de%_}{3=Fv}5J{kr s!e@QfI9>*ZZId|z+!?oR4h%@;oxCT-Wb*r1PS)booHV`7KVmtA0LmL1wEzGB delta 94 zcmZp8z}WDBQ8dXQ%+puFQqO<^2m}}yvbarC7Ht&$&CeLKSy13JKkpeP28IY0h$K)* s;j=z#GA{$e?#Y}1?u@%P2L`0_vgYR&lxR-glVUP?bu9PhU$JaL0G$^a`v3p{ diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/model.py b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/model.py index 66a00e2be..ed4f6ec1a 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/model.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/model.py @@ -89,7 +89,6 @@ class Transducer(nn.Module): prune_range: int = 5, am_scale: float = 0.0, lm_scale: float = 0.0, - prompt=None, ) -> Tuple[torch.Tensor, torch.Tensor, torch.Tensor]: """ Args: @@ -125,7 +124,7 @@ class Transducer(nn.Module): assert x.size(0) == x_lens.size(0) == y.dim0 - encoder_out, x_lens = self.encoder(x, x_lens, prompt=prompt) + encoder_out, x_lens = self.encoder(x, x_lens, prompt=self.prompt) assert torch.all(x_lens > 0) # compute ctc log-probs