rename register tables

游雁

2023-12-21 a1b0cd33d50cee3e4612d1e787399e508b453a4a

 funasr/models/bici_paraformer/cif_predictor.py

@@ -8,7 +8,7 @@
from funasr.models.scama.utils import sequence_mask

from typing import Optional, Tuple



from funasr.utils.register import register_class

from funasr.register import tables





class mae_loss(nn.Module):

@@ -94,7 +94,7 @@
    fires = torch.stack(list_fires, 1)

    return fires



@register_class("predictor_classes", "CifPredictorV3")

@tables.register("predictor_classes", "CifPredictorV3")

class CifPredictorV3(nn.Module):

    def __init__(self,

                 idim,