zhifu gao
2024-06-24 abb33d6b2097e5b0643326bc1b376a63cdc2f967
funasr/models/transformer/scorers/length_bonus.py
@@ -1,4 +1,5 @@
"""Length bonus module."""
from typing import Any
from typing import List
from typing import Tuple
@@ -54,8 +55,6 @@
        """
        return (
            torch.tensor([1.0], device=xs.device, dtype=xs.dtype).expand(
                ys.shape[0], self.n
            ),
            torch.tensor([1.0], device=xs.device, dtype=xs.dtype).expand(ys.shape[0], self.n),
            None,
        )