fix graph llm

ztxz16 · Jul 19, 2024 · ec2921c · ec2921c
1 parent 41ec201
commit ec2921c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/models/graphllm.cpp b/src/models/graphllm.cpp
@@ -270,7 +270,7 @@ namespace fastllm {
                 SplitBatch(logits, 1, batch, pointersK);
             } else */{
                 for (int b = 0; b < batch; b++) {
-                    Split(logits, 1, total + seqLens[b] - 1, total + seqLens[b], curLogits[b]);
+                    Split(logits, 1, b, b + 1, curLogits[b]);
                 }
             }