Merge pull request #31 from garfbargle/main

deal with llama.cpp api change
eastriverlee · Aug 18, 2024 · 4689414 · 4689414
2 parents 0441acc + 60bb5ec
commit 4689414
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/Sources/LLM/LLM.swift b/Sources/LLM/LLM.swift
@@ -379,10 +379,10 @@ extension Model {
 
     public func shouldAddBOS() -> Bool {
         let addBOS = llama_add_bos_token(self);
-        guard addBOS != -1 else {
+        guard !addBOS else {
             return llama_vocab_type(self) == LLAMA_VOCAB_TYPE_SPM
         }
-        return addBOS != 0
+        return addBOS
     }
 
     public func decodeOnly(_ token: Token) -> String {