expose eosToken

2024-02-26 14:58:51 -08:00
parent 4fad86d84b
commit 3f02fcc1cb
2 changed files with 51 additions and 3 deletions
--- a/Tools/llm-tool/LLMTool.swift
+++ b/Tools/llm-tool/LLMTool.swift
@@ -64,7 +64,7 @@ struct SyncGenerator: AsyncParsableCommand {
            }

            let t = token.item(Int.self)
-            if t == tokenizer.unknownTokenId {
+            if t == tokenizer.unknownTokenId || t == tokenizer.eosTokenId {
                break
            }
            tokens.append(t)
@@ -150,7 +150,7 @@ struct AsyncGenerator: AsyncParsableCommand {
                start = now
            }

-            if token == tokenizer.unknownTokenId {
+            if token == tokenizer.unknownTokenId || token == tokenizer.eosTokenId {
                break
            }
            tokens.append(token)