@@ -40,14 +40,16 @@ class TokenJsonConfig final {
40
40
tokenizer_class_ = json_config.value (" tokenizer_class" , " " );
41
41
42
42
auto tok_iter = json_config.find (" bos_token" );
43
- if (tok_iter != json_config.end () && tok_iter->is_object ()) {
44
- bos_token_ = tok_iter->value (" content" , " " );
45
- eos_token_ = json_config.value (" /eos_token/content" _json_pointer, " " );
46
- unk_token_ = json_config.value (" /unk_token/content" _json_pointer, " " );
47
- } else {
48
- bos_token_ = json_config.value (" bos_token" , " " );
49
- eos_token_ = json_config.value (" eos_token" , " " );
50
- unk_token_ = json_config.value (" unk_token" , " " );
43
+ if (tok_iter != json_config.end () && !tok_iter->is_null ()) {
44
+ if (tok_iter->is_object ()) {
45
+ bos_token_ = tok_iter->value (" content" , " " );
46
+ eos_token_ = json_config.value (" /eos_token/content" _json_pointer, " " );
47
+ unk_token_ = json_config.value (" /unk_token/content" _json_pointer, " " );
48
+ } else {
49
+ bos_token_ = json_config.value (" bos_token" , " " );
50
+ eos_token_ = json_config.value (" eos_token" , " " );
51
+ unk_token_ = json_config.value (" unk_token" , " " );
52
+ }
51
53
}
52
54
53
55
auto pad_iter = json_config.find (" pad_token" );
@@ -62,9 +64,7 @@ class TokenJsonConfig final {
62
64
return {};
63
65
}
64
66
65
- const std::string& GetVocabDataFile () const {
66
- return vocab_path_;
67
- }
67
+ const std::string& GetVocabDataFile () const { return vocab_path_; }
68
68
69
69
public:
70
70
bool add_bos_token_{};
0 commit comments