Merge pull request #855 from alibaba-damo-academy/dev_onnx

Xian Shi

2023-10-17 0dd01a880b8a7ed1d4549ee9b3a3d224e6fa1409

 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)