llama_model_n_params

This commit is contained in:
FirstTimeEZ 2024-11-15 03:11:35 +13:00 committed by GitHub
parent 0a7fecc17c
commit d205ee9273
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -472,7 +472,7 @@ extern "C" {
LLAMA_API size_t llama_model_size(const struct llama_model * model);
// Returns the total number of parameters in the model
LLAMA_API int64_t llama_model_n_params(const struct llama_model * model);
LLAMA_API uint64_t llama_model_n_params(const struct llama_model * model);
// Get a llama model tensor
LLAMA_API struct ggml_tensor * llama_get_model_tensor(struct llama_model * model, const char * name);