from local

This commit is contained in:
dohe0342 2023-02-25 15:51:30 +09:00
parent c599a30b07
commit 3d87d438fc
3 changed files with 1 additions and 0 deletions

View File

@ -263,6 +263,7 @@ class Transformer(nn.Module):
sos_id: int,
eos_id: int,
warmup: float = 1.0,
return_output: bool = False,
) -> torch.Tensor:
"""
Args: