Skip to content

Commit 6d233bc

Browse files
author
jaime-m-p
committed
Remove previous space
1 parent 0cc6593 commit 6d233bc

File tree

2 files changed

+4
-3
lines changed

2 files changed

+4
-3
lines changed

examples/batched.swift/Sources/main.swift

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -229,7 +229,7 @@ private func tokenize(text: String, add_bos: Bool) -> [llama_token] {
229229

230230
private func token_to_piece(token: llama_token, buffer: inout [CChar]) -> String? {
231231
var result = [CChar](repeating: 0, count: 8)
232-
let nTokens = llama_token_to_piece(model, token, &result, Int32(result.count), false)
232+
let nTokens = llama_token_to_piece(model, token, &result, Int32(result.count), 0, false)
233233
if nTokens < 0 {
234234
let actualTokensCount = -Int(nTokens)
235235
result = .init(repeating: 0, count: actualTokensCount)
@@ -238,6 +238,7 @@ private func token_to_piece(token: llama_token, buffer: inout [CChar]) -> String
238238
token,
239239
&result,
240240
Int32(result.count),
241+
0,
241242
false
242243
)
243244
assert(check == actualTokensCount)

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

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -322,15 +322,15 @@ actor LlamaContext {
322322
defer {
323323
result.deallocate()
324324
}
325-
let nTokens = llama_token_to_piece(model, token, result, 8, false)
325+
let nTokens = llama_token_to_piece(model, token, result, 8, 0, false)
326326

327327
if nTokens < 0 {
328328
let newResult = UnsafeMutablePointer<Int8>.allocate(capacity: Int(-nTokens))
329329
newResult.initialize(repeating: Int8(0), count: Int(-nTokens))
330330
defer {
331331
newResult.deallocate()
332332
}
333-
let nNewTokens = llama_token_to_piece(model, token, newResult, -nTokens, false)
333+
let nNewTokens = llama_token_to_piece(model, token, newResult, -nTokens, 0, false)
334334
let bufferPointer = UnsafeBufferPointer(start: newResult, count: Int(nNewTokens))
335335
return Array(bufferPointer)
336336
} else {

0 commit comments

Comments
 (0)