tp_inference device bug

shixian.shi

2023-03-09 c441eb08c44dfd4a7a8c68970fd3ebe7943d06ee

tp_inference device bug

 funasr/bin/tp_inference.py

@@ -112,6 +112,9 @@
        tp_model, tp_train_args = ASRTask.build_model_from_file(
            timestamp_infer_config, timestamp_model_file, device
        )
        if 'cuda' in device:
            tp_model = tp_model.cuda()
            
        frontend = None
        if tp_train_args.frontend is not None:
            frontend = WavFrontend(cmvn_file=timestamp_cmvn_file, **tp_train_args.frontend_conf)
@@ -240,7 +243,6 @@
        device = "cuda"
    else:
        device = "cpu"

    # 1. Set random-seed
    set_all_random_seed(seed)