funasr/models/predictor/cif.py
@@ -4,7 +4,7 @@ from funasr.modules.nets_utils import make_pad_mask class CifPredictor(nn.Module): def __init__(self, idim, l_order, r_order, threshold=1.0, dropout=0.1, smooth_factor=1.0, noise_threshold=0): def __init__(self, idim, l_order, r_order, threshold=1.0, dropout=0.1, smooth_factor=1.0, noise_threshold=0, tail_threshold=0.45): super(CifPredictor, self).__init__() self.pad = nn.ConstantPad1d((l_order, r_order), 0)
