Przeglądaj źródła

Update asr_inference_launch.py

hnluo 2 lat temu
rodzic
commit
0f18a6d2a0
1 zmienionych plików z 0 dodań i 2 usunięć
  1. 0 2
      funasr/bin/asr_inference_launch.py

+ 0 - 2
funasr/bin/asr_inference_launch.py

@@ -260,8 +260,6 @@ def inference_paraformer(
         hotword_list_or_file = None
         clas_scale = 1.0
 
-    if kwargs.get("device", None) == "cpu":
-        ngpu = 0
     if ngpu >= 1 and torch.cuda.is_available():
         device = "cuda"
     else: