Przeglądaj źródła

Merge pull request #858 from alancucki/jasper_inference_fix

Fix inference assert bug with wav list
nv-kkudrynski 5 lat temu
rodzic
commit
3081680af8
1 zmienionych plików z 5 dodań i 2 usunięć
  1. 5 2
      PyTorch/SpeechRecognition/Jasper/inference.py

+ 5 - 2
PyTorch/SpeechRecognition/Jasper/inference.py

@@ -212,8 +212,11 @@ def main():
 
     # dataset
     if args.transcribe_wav or args.transcribe_filelist:
-        assert not use_dali, "DALI is not supported for a single audio"
-        assert not args.transcribe_filelist
+
+        if use_dali:
+            print("DALI supported only with input .json files; disabling")
+            use_dali = False
+
         assert not args.pad_to_max_duration
         assert not (args.transcribe_wav and args.transcribe_filelist)