apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-12-24 05:00:08 +08:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

[Model] Fix model format of uie (#1218)

Browse Source 
Fix model format of uie
This commit is contained in:
Jack Zhou
2023-02-02 10:34:27 +08:00
committed by GitHub
parent 126c1a222a
commit 147cb2c32e
1 changed files with 18 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
fastdeploy/text/uie/model.cc
View File

@@ -13,8 +13,7 @@
// limitations under the License.
#include "fastdeploy/text/uie/model.h"
#include "fastdeploy/function/concat.h"
#include "fastdeploy/function/split.h"
#include <algorithm>
#include <codecvt>
#include <locale>
@@ -23,6 +22,8 @@
#include "fast_tokenizer/pretokenizers/pretokenizer.h"
#include "fast_tokenizer/utils/utf8.h"
#include "fastdeploy/function/concat.h"
#include "fastdeploy/function/split.h"
namespace fastdeploy {
namespace text {
@@ -169,12 +170,13 @@ UIEModel::UIEModel(const std::string& model_file,
const fastdeploy::RuntimeOption& custom_option,
const fastdeploy::ModelFormat& model_format,
SchemaLanguage schema_language)
: max_length_(max_length), position_prob_(position_prob),
schema_language_(schema_language), batch_size_(batch_size),
: max_length_(max_length),
position_prob_(position_prob),
schema_language_(schema_language),
batch_size_(batch_size),
tokenizer_(vocab_file) {
runtime_option = custom_option;
runtime_option.model_format = model_format;
runtime_option.SetModelPath(model_file, params_file);
runtime_option.SetModelPath(model_file, params_file, model_format);
initialized = Initialize();
SetSchema(schema);
tokenizer_.EnableTruncMethod(
@@ -190,12 +192,13 @@ UIEModel::UIEModel(const std::string& model_file,
const fastdeploy::RuntimeOption& custom_option,
const fastdeploy::ModelFormat& model_format,
SchemaLanguage schema_language)
: max_length_(max_length), position_prob_(position_prob),
schema_language_(schema_language), batch_size_(batch_size),
: max_length_(max_length),
position_prob_(position_prob),
schema_language_(schema_language),
batch_size_(batch_size),
tokenizer_(vocab_file) {
runtime_option = custom_option;
runtime_option.model_format = model_format;
runtime_option.SetModelPath(model_file, params_file);
runtime_option.SetModelPath(model_file, params_file, model_format);
initialized = Initialize();
SetSchema(schema);
tokenizer_.EnableTruncMethod(
@@ -210,12 +213,13 @@ UIEModel::UIEModel(const std::string& model_file,
const fastdeploy::RuntimeOption& custom_option,
const fastdeploy::ModelFormat& model_format,
SchemaLanguage schema_language)
: max_length_(max_length), position_prob_(position_prob),
schema_language_(schema_language), batch_size_(batch_size),
: max_length_(max_length),
position_prob_(position_prob),
schema_language_(schema_language),
batch_size_(batch_size),
tokenizer_(vocab_file) {
runtime_option = custom_option;
runtime_option.model_format = model_format;
runtime_option.SetModelPath(model_file, params_file);
runtime_option.SetModelPath(model_file, params_file, model_format);
initialized = Initialize();
SetSchema(schema);
tokenizer_.EnableTruncMethod(