修复为支持新版本的热词 (#1137)

夜雨飘零

2023-11-30 5eabdd444ea07038b6e814a022a212e8a87f6a9a

 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)