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)