Update ggml.c

This commit is contained in:
Georgi Gerganov 2024-02-05 13:13:12 +02:00 committed by GitHub
parent 43975144d6
commit 522660eaa3
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

2
ggml.c
View file

@ -11889,7 +11889,7 @@ GGML_CALL void ggml_rope_yarn_corr_dims(
) {
// start and end correction dims
float start = floorf(ggml_rope_yarn_corr_dim(n_dims, n_orig_ctx, beta_fast, freq_base));
float end = ceilf(ggml_rope_yarn_corr_dim(n_dims, n_orig_ctx, beta_slow, freq_base));
float end = ceilf(ggml_rope_yarn_corr_dim(n_dims, n_orig_ctx, beta_slow, freq_base));
dims[0] = MAX(0, start);
dims[1] = MIN(n_dims - 1, end);
}