smohan-speech
2023-05-11 e6d49d086f481465f6e0ce6dc68cc8afbab266e2
funasr/bin/asr_inference_paraformer.py
@@ -631,7 +631,9 @@
        export_mode = param_dict.get("export_mode", False)
    else:
        hotword_list_or_file = None
    if kwargs.get("device", None) == "cpu":
        ngpu = 0
    if ngpu >= 1 and torch.cuda.is_available():
        device = "cuda"
    else: