From 173ebc78782b2eaed40b3b29989502cb6ecdea44 Mon Sep 17 00:00:00 2001 From: Andrei Betlen Date: Thu, 25 Apr 2024 21:36:09 -0400 Subject: [PATCH] fix: Remove duplicate pooling_type definition and add misisng n_vocab definition in bindings --- llama_cpp/llama_cpp.py | 6 ------ 1 file changed, 6 deletions(-) diff --git a/llama_cpp/llama_cpp.py b/llama_cpp/llama_cpp.py index 609dd36..3b96adc 100644 --- a/llama_cpp/llama_cpp.py +++ b/llama_cpp/llama_cpp.py @@ -1189,12 +1189,6 @@ def llama_rope_type(model: llama_model_p, /) -> int: ... -# LLAMA_API enum llama_pooling_type llama_pooling_type(const struct llama_model * model); -@ctypes_function("llama_pooling_type", [llama_model_p_ctypes], ctypes.c_int) -def llama_pooling_type(model: llama_model_p, /) -> int: - ... - - # LLAMA_API int32_t llama_n_vocab (const struct llama_model * model); @ctypes_function("llama_n_vocab", [llama_model_p_ctypes], ctypes.c_int32) def llama_n_vocab(model: llama_model_p, /) -> int: