Change Emu2 template to contain 64 image tokens

kushal-10 · Apr 30, 2024 · 31f02f9 · 31f02f9
1 parent 92c6a00
commit 31f02f9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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]<image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image>[/IMG] {{ message['content'] }}{% else %}[ASSISTANT]: {{ message['content'] }} {% endif %}{% endfor %}[ASSISTANT]:"
+    "custom_chat_template": "{%- for message in messages -%}{% if message['role'] == 'user' %}[USER]: [IMG]<image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image><image>[/IMG] {{ message['content'] }}{% else %}[ASSISTANT]: {{ message['content'] }} {% endif %}{% endfor %}[ASSISTANT]:"
   }
 ]