batch.seq_id[j] -> batch.seq_id[j][0]

This commit is contained in:
Galunid 2023-10-22 13:05:05 +02:00
parent db09c028b4
commit 01533762b8

View file

@ -5990,7 +5990,7 @@ static struct ggml_cgraph * llm_build_stablelm(
for (int h = 0; h < 1; ++h) {
for (int j = 0; j < n_tokens; ++j) {
const llama_pos pos = batch.pos[j];
const llama_seq_id seq_id = batch.seq_id[j];
const llama_seq_id seq_id = batch.seq_id[j][0];
for (int i = 0; i < n_kv; ++i) {
if (!kv_self.cells[i].has_seq_id(seq_id) || kv_self.cells[i].pos > pos) {