Просмотр исходного кода

Dev bat modelscope2 (#757)

* remove from_pretrained

* bug fix for bat infer

---------

Co-authored-by: aky15 <ankeyu.aky@11.17.44.249>
aky15 2 лет назад
Родитель
Сommit
b5ed957ae9
1 измененных файлов с 1 добавлено и 1 удалено
  1. 1 1
      funasr/bin/asr_inference_launch.py

+ 1 - 1
funasr/bin/asr_inference_launch.py

@@ -1340,7 +1340,7 @@ def inference_transducer(
         format="%(asctime)s (%(module)s:%(lineno)d) %(levelname)s: %(message)s",
     )
 
-    if ngpu >= 1:
+    if ngpu >= 1 and torch.cuda.is_available():
         device = "cuda"
     else:
         device = "cpu"