zhifu gao
2024-01-24 e4035edb4620e483279923be3de9af9c35b9ce67
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: