funasr/models/e2e_asr_paraformer.py @@ -978,6 +978,7 @@ loss, stats, weight = force_gatherable((loss, stats, batch_size), loss.device) return loss, stats, weight class ContextualParaformer(Paraformer): """ Paraformer model with contextual hotword