Shi Xian
2023-12-06 b3fcd42bf6e60372ed0f0d3e53798fcdd713fe08
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)