fix: fix small detail
This commit is contained in:
parent
795ff1d3d3
commit
d6ac931b7a
1 changed files with 1 additions and 1 deletions
2
ggml.c
2
ggml.c
|
@ -12258,7 +12258,7 @@ static void ggml_compute_forward_soft_max_f32(
|
||||||
const float slope = h < n_head_log2 ? powf(m0, h + 1) : powf(m1, 2*(h - n_head_log2) + 1);
|
const float slope = h < n_head_log2 ? powf(m0, h + 1) : powf(m1, 2*(h - n_head_log2) + 1);
|
||||||
|
|
||||||
for (int i = 0; i < nc; i++) {
|
for (int i = 0; i < nc; i++) {
|
||||||
if (pos == NULL) {
|
if (pos != NULL) {
|
||||||
wp[i] = wp[i] + pos[i];
|
wp[i] = wp[i] + pos[i];
|
||||||
} else {
|
} else {
|
||||||
wp[i] = wp[i] - slope*abs(i1%nc - i);
|
wp[i] = wp[i] - slope*abs(i1%nc - i);
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue