Skip to content

Commit 516db43

Browse files
authored
Revert "Handle null bos and eos token (#66)"
This reverts commit c11169c.
1 parent bdda7c8 commit 516db43

File tree

1 file changed

+2
-9
lines changed

1 file changed

+2
-9
lines changed

src/hf_tokenizer.cpp

Lines changed: 2 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -147,15 +147,8 @@ Error HFTokenizer::load(const std::string& path) {
147147

148148
// Pull out the token strings
149149
try {
150-
const std::string bos_token =
151-
parsed_config_json.contains("bos_token") && !parsed_config_json["bos_token"].is_null()
152-
? parsed_config_json["bos_token"].get<std::string>()
153-
: "";
154-
155-
const std::string eos_token =
156-
parsed_config_json.contains("eos_token") && !parsed_config_json["eos_token"].is_null()
157-
? parsed_config_json["eos_token"].get<std::string>()
158-
: "";
150+
const std::string bos_token = parsed_config_json.at("bos_token");
151+
const std::string eos_token = parsed_config_json.at("eos_token");
159152
const auto bos_res = special_token_map_->tryGetInteger(bos_token);
160153
const auto eos_res = special_token_map_->tryGetInteger(eos_token);
161154
if (!bos_res) {

0 commit comments

Comments
 (0)