Преглед изворни кода

Dev bat modelscope2 (#757)

* remove from_pretrained

* bug fix for bat infer

---------

Co-authored-by: aky15 <ankeyu.aky@11.17.44.249>
aky15 пре 2 година
родитељ
комит
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"