from local

This commit is contained in:
dohe0342 2023-01-09 19:25:11 +09:00
parent 1c11c19869
commit 04800893e1
2 changed files with 1 additions and 1 deletions

View File

@ -188,7 +188,7 @@ class Conformer(EncoderInterface):
warmup=warmup,
) # (T, N, C)
else:
x = self.encoder(
x, layer_outputs = self.encoder(
x,
pos_emb,
src_key_padding_mask=src_key_padding_mask,