From 7b75e9de2d54ea2dcb522c91f3c615826239e074 Mon Sep 17 00:00:00 2001
From: Ettore Di Giacinto <mudler@users.noreply.github.com>
Date: Thu, 28 Nov 2024 09:34:35 +0100
Subject: [PATCH] fix(rwkv model): add stoptoken (#4283)

Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
---
 gallery/rwkv.yaml               | 1 +
 tests/models_fixtures/rwkv.yaml | 1 +
 2 files changed, 2 insertions(+)

diff --git a/gallery/rwkv.yaml b/gallery/rwkv.yaml
index 41dfcfad..68693799 100644
--- a/gallery/rwkv.yaml
+++ b/gallery/rwkv.yaml
@@ -16,6 +16,7 @@ config_file: |
 
     stopwords:
     - 'Assistant:'
+    - '<s>'
 
     template:
       chat: "{{.Input}}\nAssistant: "
diff --git a/tests/models_fixtures/rwkv.yaml b/tests/models_fixtures/rwkv.yaml
index bf54394f..f66cfe21 100644
--- a/tests/models_fixtures/rwkv.yaml
+++ b/tests/models_fixtures/rwkv.yaml
@@ -14,6 +14,7 @@ roles:
 
 stopwords:
 - 'Assistant:'
+- '<s>'
 
 template:
   chat: |