funasr/models/frontend/fused.py
@@ -1,3 +1,4 @@ from funasr.models.frontend.abs_frontend import AbsFrontend from funasr.models.frontend.default import DefaultFrontend from funasr.models.frontend.s3prl import S3prlFrontend import numpy as np @@ -6,7 +7,7 @@ from typing import Tuple class FusedFrontends(torch.nn.Module): class FusedFrontends(AbsFrontend): def __init__( self, frontends=None, align_method="linear_projection", proj_dim=100, fs=16000 ): @@ -142,4 +143,4 @@ else: raise NotImplementedError return input_feats, feats_lens return input_feats, feats_lens