From c9f5b17f1fbfaa3647702496893ebd1a4204714e Mon Sep 17 00:00:00 2001
From: Grail Finder <wohilas@gmail.com>
Date: Sun, 16 Feb 2025 09:22:15 +0300
Subject: Feat: divide continue-gen and next-msg-gen

---
 models/models.go | 1 +
 1 file changed, 1 insertion(+)

(limited to 'models/models.go')

diff --git a/models/models.go b/models/models.go
index ceb98fd..bb61abf 100644
--- a/models/models.go
+++ b/models/models.go
@@ -181,6 +181,7 @@ func NewLCPReq(prompt string, cfg *config.Config, props map[string]float32) Llam
 		Stop: []string{
 			cfg.UserRole + ":\n", "<|im_end|>",
 			cfg.ToolRole + ":\n",
+			cfg.AssistantRole + ":\n",
 		},
 	}
 }
-- 
cgit v1.2.3