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