fp16

游雁

2024-06-11 63e60cc43ddab5d28908e5e84e26a0553eb120f8

 funasr/models/transformer/scorers/scorer_interface.py

@@ -51,9 +51,7 @@
        """
        return None if state is None else state[i]

    def score(
        self, y: torch.Tensor, state: Any, x: torch.Tensor
    ) -> Tuple[torch.Tensor, Any]:
    def score(self, y: torch.Tensor, state: Any, x: torch.Tensor) -> Tuple[torch.Tensor, Any]:
        """Score new token (required).

        Args: