fix: Fix bad seq_id[i] indexing when using split_equal with unified cache

Branch: GraniteFour

Signed-off-by: Gabe Goodhart <[email protected]>

Author: gabe-l-hart

src/llama-kv-cache.cpp

@@ -579,7 +579,7 @@ int32_t llama_kv_cache_unified::find_slot(const llama_ubatch & ubatch) const {
         bool found = true;
         for (uint32_t i = 0; i < n_tokens; i++) {
             const llama_pos    pos    = ubatch.pos[i];
-            const llama_seq_id seq_id = ubatch.seq_id[i][0];
+            const llama_seq_id seq_id = ubatch.seq_id[i] == nullptr ? ubatch.seq_id[0][0] : ubatch.seq_id[i][0];
 
             // can we use this cell? either:
             //  - the cell is empty