From 7a3ab0ccef86539817e6e543e08fae1f85814a85 Mon Sep 17 00:00:00 2001
From: eastriver <eastriver0720@gmail.com>
Date: Mon, 29 Jan 2024 20:29:21 +0900
Subject: [PATCH] fix initializer with template

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

diff --git a/Sources/LLM/LLM.swift b/Sources/LLM/LLM.swift
index d66e8c6..36b65dd 100644
--- 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