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

Rename pybind/fastdeploy_runtime.cc to pybind/runtime.cc (#273)

Browse Source 
* fix yaml

* Rename pybind/fastdeploy_runtime.cc to runtime.cc
This commit is contained in:
Jason
2022-09-23 11:16:02 +08:00
committed by GitHub
parent efa7411ebb
commit 2d3e61582d
7 changed files with 11 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

0
fastdeploy/pybind/fastdeploy_runtime.cc → fastdeploy/pybind/runtime.cc
View File

8
fastdeploy/runtime.cc
View File

@@ -139,16 +139,16 @@ ModelFormat GuessModelFormat(const std::string& model_file) {
void RuntimeOption::SetModelPath(const std::string& model_path,
const std::string& params_path,
const std::string& _model_format) {
if (_model_format == "paddle") {
const ModelFormat& format) {
if (format == ModelFormat::PADDLE) {
model_file = model_path;
params_file = params_path;
model_format = ModelFormat::PADDLE;
} else if (_model_format == "onnx") {
} else if (format == ModelFormat::ONNX) {
model_file = model_path;
model_format = ModelFormat::ONNX;
} else {
FDASSERT(false, "The model format only can be 'paddle' or 'onnx'.");
FDASSERT(false, "The model format only can be ModelFormat::PADDLE/ModelFormat::ONNX.");
}
}

2
fastdeploy/runtime.h
View File

@@ -43,7 +43,7 @@ struct FASTDEPLOY_DECL RuntimeOption {
// model_format support 'paddle' / 'onnx' now.
void SetModelPath(const std::string& model_path,
const std::string& params_path = "",
const std::string& _model_format = "paddle");
const ModelFormat& format = ModelFormat::PADDLE);
// set model inference in GPU
void UseCpu();