zhifu gao
2024-01-22 d8ef0e9eb438ac8e5264b7868e29b47588f4be3b
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: