From 31f02f9f1391d154a684db6222d27e405041740f Mon Sep 17 00:00:00 2001 From: Gnurro Date: Tue, 30 Apr 2024 18:25:21 +0200 Subject: [PATCH] Change Emu2 template to contain 64 image tokens --- backends/model_registry.json | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/backends/model_registry.json b/backends/model_registry.json index 37558f0d68..ec111313f0 100644 --- a/backends/model_registry.json +++ b/backends/model_registry.json @@ -449,6 +449,6 @@ "eos_to_cull": "[Assistant]:", "padding": false, "premade_chat_template": false, - "custom_chat_template": "{%- for message in messages -%}{% if message['role'] == 'user' %}[USER]: [IMG][/IMG] {{ message['content'] }}{% else %}[ASSISTANT]: {{ message['content'] }} {% endif %}{% endfor %}[ASSISTANT]:" + "custom_chat_template": "{%- for message in messages -%}{% if message['role'] == 'user' %}[USER]: [IMG][/IMG] {{ message['content'] }}{% else %}[ASSISTANT]: {{ message['content'] }} {% endif %}{% endfor %}[ASSISTANT]:" } ] \ No newline at end of file