summaryrefslogtreecommitdiff
path: root/models
diff options
context:
space:
mode:
authorGrail Finder <wohilas@gmail.com>2025-02-16 09:22:15 +0300
committerGrail Finder <wohilas@gmail.com>2025-02-16 09:22:15 +0300
commitc9f5b17f1fbfaa3647702496893ebd1a4204714e (patch)
tree14d0cf27038df57fed68fee23364a48035dd7452 /models
parentc1344794143ef48670d8eeb365a10a5295a145ae (diff)
Feat: divide continue-gen and next-msg-gen
Diffstat (limited to 'models')
-rw-r--r--models/models.go1
1 files changed, 1 insertions, 0 deletions
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",
},
}
}