Skip to content

Commit eda54a4

Browse files
committed
feat: add LLAMA_VOCAB_PRE_TYPE_JAIS constant
1 parent 4e5eb9c commit eda54a4

File tree

3 files changed

+5
-0
lines changed

3 files changed

+5
-0
lines changed

ext/llama_cpp/dummy.rb

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -94,6 +94,9 @@ module LLaMACpp # rubocop:disable Metrics/ModuleLength
9494
# LLaMA vocabulary pre-tokenization type.
9595
LLAMA_VOCAB_PRE_TYPE_VIKING = 16
9696

97+
# LLaMA vocabulary pre-tokenization type.
98+
LLAMA_VOCAB_PRE_TYPE_JAIS = 17
99+
97100
# LLaMA token attribute type.
98101
LLAMA_TOKEN_ATTR_UNDEFINED = 0
99102

ext/llama_cpp/llama_cpp.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3514,6 +3514,7 @@ extern "C" void Init_llama_cpp(void) {
35143514
rb_define_const(rb_mLLaMACpp, "LLAMA_VOCAB_PRE_TYPE_SMAUG", INT2NUM(LLAMA_VOCAB_PRE_TYPE_SMAUG));
35153515
rb_define_const(rb_mLLaMACpp, "LLAMA_VOCAB_PRE_TYPE_PORO", INT2NUM(LLAMA_VOCAB_PRE_TYPE_PORO));
35163516
rb_define_const(rb_mLLaMACpp, "LLAMA_VOCAB_PRE_TYPE_VIKING", INT2NUM(LLAMA_VOCAB_PRE_TYPE_VIKING));
3517+
rb_define_const(rb_mLLaMACpp, "LLAMA_VOCAB_PRE_TYPE_JAIS", INT2NUM(LLAMA_VOCAB_PRE_TYPE_JAIS));
35173518

35183519
rb_define_const(rb_mLLaMACpp, "LLAMA_TOKEN_TYPE_UNDEFINED", INT2NUM(LLAMA_TOKEN_TYPE_UNDEFINED));
35193520
rb_define_const(rb_mLLaMACpp, "LLAMA_TOKEN_TYPE_NORMAL", INT2NUM(LLAMA_TOKEN_TYPE_NORMAL));

sig/llama_cpp.rbs

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -34,6 +34,7 @@ module LLaMACpp
3434
LLAMA_VOCAB_PRE_TYPE_SMAUG: Integer
3535
LLAMA_VOCAB_PRE_TYPE_PORO: Integer
3636
LLAMA_VOCAB_PRE_TYPE_VIKING: Integer
37+
LLAMA_VOCAB_PRE_TYPE_JAIS: Integer
3738

3839
LLAMA_TOKEN_ATTR_UNDEFINED: Integer
3940
LLAMA_TOKEN_ATTR_UNKNOWN: Integer

0 commit comments

Comments
 (0)