shixian.shi 2 роки тому
батько
коміт
472c0f9265
1 змінених файлів з 1 додано та 1 видалено
  1. 1 1
      funasr/datasets/large_datasets/dataset.py

+ 1 - 1
funasr/datasets/large_datasets/dataset.py

@@ -101,7 +101,7 @@ class AudioDataset(IterableDataset):
                 if data_type == "kaldi_ark":
                     ark_reader = ReadHelper('ark:{}'.format(data_file))
                     reader_list.append(ark_reader)
-                elif data_type == "text" or data_type == "sound":
+                elif data_type == "text" or data_type == "sound" or data_type == 'text_hotword':
                     text_reader = open(data_file, "r")
                     reader_list.append(text_reader)
                 elif data_type == "none":