from local

This commit is contained in:
dohe0342 2023-01-09 20:02:47 +09:00
parent b926cb0045
commit fc36007b5b
2 changed files with 1 additions and 3 deletions

View File

@ -497,10 +497,8 @@ def get_transducer_model(params: AttributeDict) -> nn.Module:
)
return model
def get_interformer_model(params: AttributeDict) -> nn.Module:
def get_interformer_model(encoder, params: AttributeDict) -> nn.Module:
encoder = get_encoder_model(params)
decoder = get_decoder_model(params)
joiner = get_joiner_model(params)
model = Transducer(
encoder=encoder,