shixian.shi пре 2 година
родитељ
комит
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]
                             result[k] = restored_data[j][k]
                         else:
                         else:
                             result[k] = torch.cat([result[k], restored_data[j][k]], dim=0)
                             result[k] = torch.cat([result[k], restored_data[j][k]], dim=0)
-                    elif k == 'text':
+                    elif k == 'raw_text':
                         if k not in result:
                         if k not in result:
                             result[k] = restored_data[j][k]
                             result[k] = restored_data[j][k]
                         else:
                         else: