PhonePaint / espnet2 /enh /decoder /abs_decoder.py
Tom-Brenner's picture
PhonePaint Space: MAPS-only, ZeroGPU, assets fetched at startup (part 2)
fa1da78 verified
Raw History Blame Contribute Delete
319 Bytes
from abc import ABC
from abc import abstractmethod
from typing import Tuple
import torch
class AbsDecoder(torch.nn.Module, ABC):
@abstractmethod
def forward(
self,
input: torch.Tensor,
ilens: torch.Tensor,
) -> Tuple[torch.Tensor, torch.Tensor]:
raise NotImplementedError