Spaces:

Tom-Brenner
/

PhonePaint

Running on Zero

PhonePaint / espnet2 /enh /decoder /null_decoder.py

PhonePaint Space: MAPS-only, ZeroGPU, assets fetched at startup (part 2)

fa1da78 verified about 2 months ago

493 Bytes

import torch

from espnet2.enh.decoder.abs_decoder import AbsDecoder


class NullDecoder(AbsDecoder):
    """Null decoder, return the same args."""

    def __init__(self):
        super().__init__()

    def forward(self, input: torch.Tensor, ilens: torch.Tensor):
        """Forward. The input should be the waveform already.
        Args:
            input (torch.Tensor): wav [Batch, sample]
            ilens (torch.Tensor): input lengths [Batch]
        """
        return input, ilens

	import torch

	from espnet2.enh.decoder.abs_decoder import AbsDecoder


	class NullDecoder(AbsDecoder):
	"""Null decoder, return the same args."""

	def __init__(self):
	super().__init__()

	def forward(self, input: torch.Tensor, ilens: torch.Tensor):
	"""Forward. The input should be the waveform already.

	Args:
	input (torch.Tensor): wav [Batch, sample]
	ilens (torch.Tensor): input lengths [Batch]
	"""
	return input, ilens