Don't crash on ftype (formerly f16) == 4 (#917)

This commit is contained in:
Stephan Walter
2023-04-12 15:06:16 +00:00
committed by GitHub
parent f76cb3a34d
commit e7f6997f89
2 changed files with 4 additions and 1 deletions

View File

@@ -71,6 +71,7 @@ extern "C" {
LLAMA_FTYPE_MOSTLY_F16 = 1, // except 1d tensors
LLAMA_FTYPE_MOSTLY_Q4_0 = 2, // except 1d tensors
LLAMA_FTYPE_MOSTLY_Q4_1 = 3, // except 1d tensors
LLAMA_FTYPE_MOSTLY_Q4_1_SOME_F16 = 4, // tok_embeddings.weight and output.weight are F16
};
LLAMA_API struct llama_context_params llama_context_default_params();