From 3988004aeae58eae25b22267be34e8d7280ede9f Mon Sep 17 00:00:00 2001
From: eastriver <eastriver0720@gmail.com>
Date: Sat, 3 Feb 2024 06:40:04 +0900
Subject: [PATCH] remove redundant BOS token

---
 Sources/LLM/LLM.swift | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/Sources/LLM/LLM.swift b/Sources/LLM/LLM.swift
index c6a1862..15fe4d5 100644
--- a/Sources/LLM/LLM.swift
+++ b/Sources/LLM/LLM.swift
@@ -546,7 +546,7 @@ public struct Template {
     
     public static func llama(_ systemPrompt: String? = nil) -> Template {
         return Template(
-            prefix: "<s>[INST] ",
+            prefix: "[INST] ",
             system: ("<<SYS>>\n", "\n<</SYS>>\n\n"),
             user: ("", " [/INST]"),
             bot: (" ", "</s><s>[INST] "),
@@ -557,7 +557,6 @@ public struct Template {
     }
     
     public static let mistral = Template(
-        prefix: "<s>",
         user: ("[INST] ", " [/INST]"),
         bot: ("", "</s> "),
         stopSequence: "</s>",
@@ -672,7 +671,7 @@ extension URL {
             }
             task.resume()
         }
-        let _ = observation
+        _ = observation
         try FileManager.default.moveItem(at: url, to: destination)
     }
 }