fix initializer with template

eastriverlee · Jan 29, 2024 · 7a3ab0c · 7a3ab0c
1 parent 9665491
commit 7a3ab0c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/Sources/LLM/LLM.swift b/Sources/LLM/LLM.swift
@@ -134,6 +134,7 @@ open class LLM: ObservableObject {
     ) {
         self.init(
             from: url.path,
+            stopSequence: template.stopSequence,
             history: history,
             seed: seed,
             topK: topK,
@@ -142,7 +143,7 @@ open class LLM: ObservableObject {
             historyLimit: historyLimit,
             maxTokenCount: maxTokenCount
         )
-        self.template = template
+        self.preProcess = template.preProcess
     }
 
     private var shouldContinuePredicting = false