shixian.shi 2 лет назад
Родитель
Сommit
7ab24f4e74
1 измененных файлов с 1 добавлено и 1 удалено
  1. 1 1
      funasr/auto/auto_model.py

+ 1 - 1
funasr/auto/auto_model.py

@@ -377,7 +377,7 @@ class AutoModel:
                             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: