shixian.shi
2024-01-22 7ab24f4e746572b56c90bba30360244e873d4568
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: