diff options
Diffstat (limited to 'src/model_interface.h')
-rw-r--r-- | src/model_interface.h | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/src/model_interface.h b/src/model_interface.h index 320dc86..4a3a44e 100644 --- a/src/model_interface.h +++ b/src/model_interface.h @@ -47,10 +47,10 @@ class ModelInterface { using PieceToIdMap = std::unordered_map<absl::string_view, int, string_util::string_view_hash>; - static const char *kUNK(); - static const char *kBOS(); - static const char *kEOS(); - static const char *kPAD(); + absl::string_view unk_piece() const; + absl::string_view bos_piece() const; + absl::string_view eos_piece() const; + absl::string_view pad_piece() const; // `model_proto` should not be deleted until ModelInterface is destroyed. explicit ModelInterface(const ModelProto &model_proto); |