Skip to content

Commit 8ceda95

Browse files
committed
squash! llama : rename batch.logits to batch.output
Fix the name of the output field in the batch struct which I missed in the previous commit.
1 parent 74edb42 commit 8ceda95

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/llama.swiftui/llama.cpp.swift/LibLlama.swift

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ func llama_batch_add(_ batch: inout llama_batch, _ id: llama_token, _ pos: llama
1616
for i in 0..<seq_ids.count {
1717
batch.seq_id[Int(batch.n_tokens)]![Int(i)] = seq_ids[i]
1818
}
19-
batch.logits [Int(batch.n_tokens)] = logits ? 1 : 0
19+
batch.output [Int(batch.n_tokens)] = logits ? 1 : 0
2020

2121
batch.n_tokens += 1
2222
}

0 commit comments

Comments
 (0)