LocalAI/gallery/rwkv.yaml
Ettore Di Giacinto 7b75e9de2d
fix(rwkv model): add stoptoken (#4283)
Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
2024-11-28 09:34:35 +01:00

25 lines
379 B
YAML

---
name: "rwkv"
config_file: |
parameters:
top_k: 80
temperature: 0.9
max_tokens: 4098
top_p: 0.8
context_size: 4098
roles:
user: "User: "
system: "System: "
assistant: "Assistant: "
stopwords:
- 'Assistant:'
- '<s>'
template:
chat: "{{.Input}}\nAssistant: "
completion: |
{{.Input}}