update func cif_wo_hidden

shixian.shi

2023-08-14 c73d1a8e81582b91a9bdd6e82fce2e84f8d9d94b

update func cif_wo_hidden

 funasr/export/models/predictor/cif.py

@@ -288,7 +288,7 @@


        fire_place = integrate >= threshold

        integrate = torch.where(fire_place,

                                integrate - torch.ones([batch_size], device=alphas.device),

                                integrate - torch.ones([batch_size], device=alphas.device)*threshold,

                                integrate)



    fires = torch.stack(list_fires, 1)


 funasr/models/predictor/cif.py

@@ -499,7 +499,7 @@


        fire_place = integrate >= threshold

        integrate = torch.where(fire_place,

                                integrate - torch.ones([batch_size], device=alphas.device),

                                integrate - torch.ones([batch_size], device=alphas.device)*threshold,

                                integrate)



    fires = torch.stack(list_fires, 1)


 funasr/utils/timestamp_tools.py

@@ -19,7 +19,7 @@
        list_fires.append(integrate)
        fire_place = integrate >= threshold
        integrate = torch.where(fire_place,
                                integrate - torch.ones([batch_size], device=alphas.device),
                                integrate - torch.ones([batch_size], device=alphas.device)*threshold,
                                integrate)
    fires = torch.stack(list_fires, 1)
    return fires