分角色语音识别支持更多的模型

夜雨飘零

2023-12-05 18b1449d1ff06c469e54190508c4f6be05c73d85

 funasr/layers/inversible_interface.py

New file
@@ -0,0 +1,14 @@
from abc import ABC
from abc import abstractmethod
from typing import Tuple

import torch


class InversibleInterface(ABC):
    @abstractmethod
    def inverse(
        self, input: torch.Tensor, input_lengths: torch.Tensor = None
    ) -> Tuple[torch.Tensor, torch.Tensor]:
        # return output, output_lengths
        raise NotImplementedError