feat: add LLAMA_POOLING_TYPE_LAST constant

yoshoku · Jun 22, 2024 · 138ba89 · 138ba89
1 parent e453ee5
commit 138ba89
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 0 deletions.
diff --git a/ext/llama_cpp/dummy.rb b/ext/llama_cpp/dummy.rb
@@ -277,6 +277,9 @@ module LLaMACpp # rubocop:disable Metrics/ModuleLength
   # LLaMA pooling type.
   LLAMA_POOLING_TYPE_CLS = 2
 
+  # LLaMA pooling type.
+  LLAMA_POOLING_TYPE_LAST = 3
+
   # LLaMA split mode: single GPU.
   LLAMA_SPLIT_MODE_NONE = 0
 

diff --git a/ext/llama_cpp/llama_cpp.cpp b/ext/llama_cpp/llama_cpp.cpp
@@ -3572,6 +3572,7 @@ extern "C" void Init_llama_cpp(void) {
   rb_define_const(rb_mLLaMACpp, "LLAMA_POOLING_TYPE_NONE", INT2NUM(LLAMA_POOLING_TYPE_NONE));
   rb_define_const(rb_mLLaMACpp, "LLAMA_POOLING_TYPE_MEAN", INT2NUM(LLAMA_POOLING_TYPE_MEAN));
   rb_define_const(rb_mLLaMACpp, "LLAMA_POOLING_TYPE_CLS", INT2NUM(LLAMA_POOLING_TYPE_CLS));
+  rb_define_const(rb_mLLaMACpp, "LLAMA_POOLING_TYPE_LAST", INT2NUM(LLAMA_POOLING_TYPE_LAST));
 
   rb_define_const(rb_mLLaMACpp, "LLAMA_SPLIT_MODE_NONE", INT2NUM(LLAMA_SPLIT_MODE_NONE));
   rb_define_const(rb_mLLaMACpp, "LLAMA_SPLIT_MODE_LAYER", INT2NUM(LLAMA_SPLIT_MODE_LAYER));

diff --git a/sig/llama_cpp.rbs b/sig/llama_cpp.rbs
@@ -99,6 +99,7 @@ module LLaMACpp
   LLAMA_POOLING_TYPE_NONE: Integer
   LLAMA_POOLING_TYPE_MEAN: Integer
   LLAMA_POOLING_TYPE_CLS: Integer
+  LLAMA_POOLING_TYPE_LAST: Integer
 
   LLAMA_SPLIT_MODE_NONE: Integer
   LLAMA_SPLIT_MODE_LAYER: Integer