apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-06 00:57:33 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

[Bug Fix] When loading model from memory, don't need CheckModelFormat (#1073)

Browse Source 
When loading model from memory, don't need CheckModelFormat
This commit is contained in:
huangjianhui
2023-01-06 13:48:44 +08:00
committed by GitHub
parent d7a65e5c70
commit c8d2fef198
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
fastdeploy/fastdeploy_model.cc
View File

@@ -140,9 +140,11 @@ bool FastDeployModel::InitRuntimeWithSpecifiedDevice() {
} }
bool FastDeployModel::InitRuntime() { bool FastDeployModel::InitRuntime() {
if (!runtime_option.model_from_memory_) {
FDASSERT( FDASSERT(
CheckModelFormat(runtime_option.model_file, runtime_option.model_format), CheckModelFormat(runtime_option.model_file, runtime_option.model_format),
"ModelFormatCheck Failed."); "ModelFormatCheck Failed.");
}
if (runtime_initialized_) { if (runtime_initialized_) {
FDERROR << "The model is already initialized, cannot be initliazed again." FDERROR << "The model is already initialized, cannot be initliazed again."
<< std::endl; << std::endl;