diff options
Diffstat (limited to 'models/models.go')
| -rw-r--r-- | models/models.go | 1 |
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", }, } } |