Update ggml.c
This commit is contained in:
parent
43975144d6
commit
522660eaa3
1 changed files with 1 additions and 1 deletions
2
ggml.c
2
ggml.c
|
@ -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);
|
||||
}
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue