* remove from_pretrained * bug fix for bat infer --------- Co-authored-by: aky15 <ankeyu.aky@11.17.44.249>
@@ -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"