Spaces:

Tom-Brenner
/

PhonePaint

Running on Zero

PhonePaint / espnet2 /enh /encoder /null_encoder.py

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

fa1da78 verified about 2 months ago

503 Bytes

import torch

from espnet2.enh.encoder.abs_encoder import AbsEncoder


class NullEncoder(AbsEncoder):
    """Null encoder."""

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

    @property
    def output_dim(self) -> int:
        return 1

    def forward(self, input: torch.Tensor, ilens: torch.Tensor):
        """Forward.
        Args:
            input (torch.Tensor): mixed speech [Batch, sample]
            ilens (torch.Tensor): input lengths [Batch]
        """
        return input, ilens

	import torch

	from espnet2.enh.encoder.abs_encoder import AbsEncoder


	class NullEncoder(AbsEncoder):
	"""Null encoder."""

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

	@property
	def output_dim(self) -> int:
	return 1

	def forward(self, input: torch.Tensor, ilens: torch.Tensor):
	"""Forward.

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