summaryrefslogtreecommitdiff
path: root/examples/llama.swiftui
diff options
context:
space:
mode:
Diffstat (limited to 'examples/llama.swiftui')
-rw-r--r--examples/llama.swiftui/llama.cpp.swift/LibLlama.swift2
1 files changed, 1 insertions, 1 deletions
diff --git a/examples/llama.swiftui/llama.cpp.swift/LibLlama.swift b/examples/llama.swiftui/llama.cpp.swift/LibLlama.swift
index 66244382..8696b493 100644
--- a/examples/llama.swiftui/llama.cpp.swift/LibLlama.swift
+++ b/examples/llama.swiftui/llama.cpp.swift/LibLlama.swift
@@ -161,7 +161,7 @@ actor LlamaContext {
new_token_id = llama_sample_token_greedy(context, &candidates_p)
}
- if new_token_id == llama_token_eos(context) || n_cur == n_len {
+ if new_token_id == llama_token_eos(model) || n_cur == n_len {
print("\n")
let new_token_str = String(cString: temporary_invalid_cchars + [0])
temporary_invalid_cchars.removeAll()