summaryrefslogtreecommitdiff
path: root/common/common.cpp
diff options
context:
space:
mode:
authorGeorgi Gerganov <ggerganov@gmail.com>2024-01-12 13:01:56 +0200
committerGitHub <noreply@github.com>2024-01-12 13:01:56 +0200
commitf445c0e68cf8e1faca0b2aa8dfb9d48231cec301 (patch)
treea230e263424ba71153764e21fef7b92f46bd9fd9 /common/common.cpp
parent326b418b59b6d48d854c4461a2303e8ac0a311e6 (diff)
llama : fix llm_build_k_shift to use correct n_rot (#4889)
* llama : fix llm_build_k_shift to use correct n_rot ggml-ci * llama : always use hparams.n_rot for ggml_rope_custom ggml-ci * convert : fix persimmon conversion to write correct n_rot
Diffstat (limited to 'common/common.cpp')
-rw-r--r--common/common.cpp3
1 files changed, 3 insertions, 0 deletions
diff --git a/common/common.cpp b/common/common.cpp
index b2cb0e25..3aefed01 100644
--- a/common/common.cpp
+++ b/common/common.cpp
@@ -1055,6 +1055,9 @@ struct llama_model_params llama_model_params_from_gpt_params(const gpt_params &
}
static ggml_type kv_cache_type_from_str(const std::string & s) {
+ if (s == "f32") {
+ return GGML_TYPE_F32;
+ }
if (s == "f16") {
return GGML_TYPE_F16;
}