Skip to content

Commit e0cc2b4

Browse files
committed
fix compile on ubuntu
1 parent 2ad6538 commit e0cc2b4

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/llama-model.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -878,8 +878,8 @@ void llama_model::load_hparams(llama_model_loader & ml) {
878878
}
879879

880880
hparams.f_attention_scale = type == LLM_TYPE_27B
881-
? 1.0f / sqrtf(float(hparams.n_embd / hparams.n_head(0)))
882-
: 1.0f / sqrtf(float(hparams.n_embd_head_k));
881+
? 1.0f / std::sqrtf(float(hparams.n_embd / hparams.n_head(0)))
882+
: 1.0f / std::sqrtf(float(hparams.n_embd_head_k));
883883
} break;
884884
case LLM_ARCH_STARCODER2:
885885
{

0 commit comments

Comments
 (0)