Skip to content

Commit

Permalink
squash! llama : rename batch.logits to batch.output
Browse files Browse the repository at this point in the history
Fix the name of the output field in the batch struct which I missed in
the previous commit.
  • Loading branch information
danbev committed Oct 23, 2024
1 parent 74edb42 commit 8ceda95
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion examples/llama.swiftui/llama.cpp.swift/LibLlama.swift
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@ func llama_batch_add(_ batch: inout llama_batch, _ id: llama_token, _ pos: llama
for i in 0..<seq_ids.count {
batch.seq_id[Int(batch.n_tokens)]![Int(i)] = seq_ids[i]
}
batch.logits [Int(batch.n_tokens)] = logits ? 1 : 0
batch.output [Int(batch.n_tokens)] = logits ? 1 : 0

batch.n_tokens += 1
}
Expand Down

0 comments on commit 8ceda95

Please sign in to comment.