mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-11 06:55:27 +00:00
from local
This commit is contained in:
parent
af71c2e738
commit
0917bd9d79
Binary file not shown.
@ -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
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user