funasr/auto/auto_model.py
@@ -377,7 +377,7 @@ result[k] = restored_data[j][k] else: result[k] = torch.cat([result[k], restored_data[j][k]], dim=0) elif k == 'text': elif k == 'raw_text': if k not in result: result[k] = restored_data[j][k] else: @@ -411,7 +411,7 @@ else: # punc_segment sentence_list = timestamp_sentence(punc_res[0]['punc_array'], \ result['timestamp'], \ result['text']) result['raw_text']) distribute_spk(sentence_list, sv_output) result['sentence_info'] = sentence_list
