from local

This commit is contained in:
dohe0342 2023-04-11 15:38:03 +09:00
parent b201ef10b4
commit bb588c6d9b
2 changed files with 1 additions and 1 deletions

View File

@ -120,7 +120,7 @@ class Transducer(nn.Module):
assert x.size(0) == x_lens.size(0) == y.dim0
encoder_out, x_lens = self.encoder(x, x_lens)
encoder_out, x_lens = self.encoder(x, x_lens, prompt=prompt)
assert torch.all(x_lens > 0)
# compute ctc log-probs