From 1ce5a826b01ccd76938b404b3e32d1fb5a2f92b5 Mon Sep 17 00:00:00 2001 From: Ettore Di Giacinto Date: Thu, 28 Nov 2024 09:28:40 +0100 Subject: [PATCH] fix(rwkv model): add stoptoken Signed-off-by: Ettore Di Giacinto --- 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 41dfcfad4071..686937997db6 100644 --- a/gallery/rwkv.yaml +++ b/gallery/rwkv.yaml @@ -16,6 +16,7 @@ config_file: | stopwords: - 'Assistant:' + - '' template: chat: "{{.Input}}\nAssistant: " diff --git a/tests/models_fixtures/rwkv.yaml b/tests/models_fixtures/rwkv.yaml index bf54394fd771..f66cfe21110e 100644 --- a/tests/models_fixtures/rwkv.yaml +++ b/tests/models_fixtures/rwkv.yaml @@ -14,6 +14,7 @@ roles: stopwords: - 'Assistant:' +- '' template: chat: |