speech_asr 3 лет назад
Родитель
Сommit
3f2981bb8d
1 измененных файлов с 2 добавлено и 2 удалено
  1. 2 2
      funasr/tasks/diar.py

+ 2 - 2
funasr/tasks/diar.py

@@ -787,10 +787,10 @@ class EENDOLADiarTask(AbsTask):
             cls, train: bool = True, inference: bool = False
             cls, train: bool = True, inference: bool = False
     ) -> Tuple[str, ...]:
     ) -> Tuple[str, ...]:
         if not inference:
         if not inference:
-            retval = ("speech", "profile", "binary_labels")
+            retval = ("speech", )
         else:
         else:
             # Recognition mode
             # Recognition mode
-            retval = ("speech")
+            retval = ("speech", )
         return retval
         return retval
 
 
     @classmethod
     @classmethod