diff options
Diffstat (limited to 'src/bpe_model.h')
-rw-r--r-- | src/bpe_model.h | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/src/bpe_model.h b/src/bpe_model.h index 243664f..c6e1abe 100644 --- a/src/bpe_model.h +++ b/src/bpe_model.h @@ -42,6 +42,10 @@ class Model : public ModelInterface { // When alpha <= 0.0, no sampling is performed. EncodeResult SampleEncode(absl::string_view normalized, float alpha) const override; + + bool IsSampleEncodeAvailable() const override { return true; } + + bool IsNBestEncodeAvailable() const override { return false; } }; } // namespace bpe } // namespace sentencepiece |