| | |
| | | } |
| | | string punc_model_path; |
| | | string punc_config_path; |
| | | string token_file; |
| | | |
| | | punc_model_path = PathAppend(model_path.at(MODEL_DIR), MODEL_NAME); |
| | | if(model_path.find(QUANTIZE) != model_path.end() && model_path.at(QUANTIZE) == "true"){ |
| | | punc_model_path = PathAppend(model_path.at(MODEL_DIR), QUANT_MODEL_NAME); |
| | | } |
| | | punc_config_path = PathAppend(model_path.at(MODEL_DIR), PUNC_CONFIG_NAME); |
| | | token_file = PathAppend(model_path.at(MODEL_DIR), TOKEN_PATH); |
| | | |
| | | mm->InitPunc(punc_model_path, punc_config_path, thread_num); |
| | | mm->InitPunc(punc_model_path, punc_config_path, token_file, thread_num); |
| | | return mm; |
| | | } |
| | | |