from local

This commit is contained in:
dohe0342 2023-04-11 15:37:49 +09:00
parent 34715ee446
commit b201ef10b4
3 changed files with 1 additions and 0 deletions

View File

@ -84,6 +84,7 @@ 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: