zhifu gao
2024-05-14 2f27b165559cd53afab52047309ebe4ac838ebb8
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,
        )