llama.swiftui : use correct pointer for llama_token_eos (ggerganov#4797)

teleprint-me · Jan 6, 2024 · c75ca5d · c75ca5d
1 parent 96e80da
commit c75ca5d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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()