| | |
| | | string seg_dict_path; |
| | | |
| | | if(use_gpu){ |
| | | #ifdef USE_GPU |
| | | asr_handle = make_unique<ParaformerTorch>(); |
| | | #else |
| | | LOG(ERROR) <<"GPU is not supported! CPU will be used! If you want to use GPU, please add -DGPU=ON when cmake"; |
| | | asr_handle = make_unique<Paraformer>(); |
| | | use_gpu = false; |
| | | #endif |
| | | }else{ |
| | | asr_handle = make_unique<Paraformer>(); |
| | | } |
| | |
| | | if(model_path.find(QUANTIZE) != model_path.end() && model_path.at(QUANTIZE) == "true"){ |
| | | am_model_path = PathAppend(model_path.at(MODEL_DIR), TORCH_QUANT_MODEL_NAME); |
| | | } |
| | | if(model_path.find(BLADEDISC) != model_path.end() && model_path.at(BLADEDISC) == "true"){ |
| | | am_model_path = PathAppend(model_path.at(MODEL_DIR), BLADE_MODEL_NAME); |
| | | } |
| | | } |
| | | } |
| | | am_cmvn_path = PathAppend(model_path.at(MODEL_DIR), AM_CMVN_NAME); |