Skip to content

Commit a1549c0

Browse files
author
plusbang
committed
add initiate call
1 parent 2ae6078 commit a1549c0

File tree

1 file changed

+3
-7
lines changed

1 file changed

+3
-7
lines changed

src/cpp/src/module_genai/modules/md_text_encoder.cpp

Lines changed: 3 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,9 @@ namespace genai {
1313
namespace module {
1414

1515
TextEncoderModule::TextEncoderModule(const IBaseModuleDesc::PTR& desc) : IBaseModule(desc) {
16-
16+
if (!initialize()) {
17+
std::cerr << "Failed to initiate TextEncoderModule" << std::endl;
18+
}
1719
}
1820

1921
bool TextEncoderModule::initialize() {
@@ -48,12 +50,6 @@ bool TextEncoderModule::initialize(const std::filesystem::path& tokenizer_path,
4850
}
4951

5052
TokenizedInputs TextEncoderModule::run(const std::string& prompt) {
51-
// TODO: 似乎暂时缺少各模块的initiate?
52-
const auto& params = module_desc->params;
53-
auto it_path = params.find("model_path");
54-
std::filesystem::path tokenizer_path = it_path->second;
55-
m_tokenizer_impl = std::make_shared<Tokenizer::TokenizerImpl>(tokenizer_path, m_tokenization_params);
56-
5753
OPENVINO_ASSERT(m_tokenizer_impl, "TextEncoderModule is not initialized. Call initialize() first.");
5854
check_arguments(m_tokenization_params, {ov::genai::add_special_tokens.name(),
5955
ov::genai::max_length.name(),

0 commit comments

Comments
 (0)