Add tiny processor fixture for testing (ernie4_5_vl_moe)

.gitattributes

@@ -33,3 +33,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+tokenizer.json filter=lfs diff=lfs merge=lfs -text
+Roboto-Regular.ttf filter=lfs diff=lfs merge=lfs -text

Roboto-Regular.ttf

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:dbd285b518e398832f6f4a736109c355ce25a49546bfce41bab256c9ef7e56eb
+size 146004

chat_template.jinja

@@ -0,0 +1,81 @@
+
+{%- set image_count = namespace(value=0) -%}
+{%- set video_count = namespace(value=0) -%}
+{{- '<|begin_of_sentence|>' }}
+{%- for message in messages -%}
+    {%- if message.role in ['system', 'user'] -%}
+        {%- if message.role == 'user' -%}
+            {{- 'User: ' -}}
+        {%- endif -%}
+        {%- if message.content is string -%}
+            {{- message.content -}}
+        {%- else -%}
+            {%- for content_item in message.content -%}
+                {%- if content_item.type == 'text' -%}
+                    {{- content_item.text -}}
+                {%- elif content_item.type in ['image_url', 'image'] -%}
+                    {%- set image_count.value = image_count.value + 1 -%}
+                    Picture {{ image_count.value }}:<|IMAGE_START|><|IMAGE_PLACEHOLDER|><|IMAGE_END|>
+                {%- elif content_item.type in ['video_url', 'video'] -%}
+                    {%- set video_count.value = video_count.value + 1 -%}
+                    Video {{ video_count.value }}:<|VIDEO_START|><|VIDEO_PLACEHOLDER|><|VIDEO_END|>
+                {%- endif -%}
+            {%- endfor -%}
+        {%- endif -%}
+        {%- if message.role == 'system' -%}
+            {{- '
+            ' -}}
+        {%- endif -%}
+    {%- elif message.role == 'assistant' -%}
+        {%- macro extract_text_content(content_field) -%}
+            {%- if content_field is string -%}
+                {{- content_field -}}
+            {%- elif content_field is iterable and content_field is not string -%}
+                {%- set ns = namespace(text_parts=[]) -%}
+                {%- set text_parts = [] -%}
+                {%- for item in content_field -%}
+                    {%- if item.type == 'text' -%}
+                        {%- set ns.text_parts = ns.text_parts + [item.text] -%}
+                    {%- endif -%}
+                {%- endfor -%}
+                {{- ns.text_parts | join("") -}}
+            {%- else -%}
+                {{- '' -}}
+            {%- endif -%}
+        {%- endmacro -%}
+        {%- set reasoning_content = extract_text_content(message.reasoning_content) -%}
+        {%- set content = extract_text_content(message.content) -%}
+        {%- if '</think>' in content %}
+            {%- set reasoning_content = content.split('</think>')[0].rstrip('
+                        ').split('<think>')[-1].lstrip('
+                        ') %}
+            {%- set content = content.split('</think>')[-1].lstrip('
+                        ') %}
+        {%- endif %}
+        {%- if reasoning_content %}
+            {{- '
+            ' + 'Assistant: ' + '<think>
+            ' + reasoning_content.strip('
+                        ') + '
+            </think>
+            ' + content.lstrip('
+            ') }}
+        {%- else %}
+            {{- '
+            ' + 'Assistant: ' + content }}
+        {%- endif %}
+        {{- '<|end_of_sentence |>' }}
+    {%- endif -%}
+{%- endfor -%}
+{%- if add_generation_prompt is not defined or add_generation_prompt is true %}
+    {{- '
+Assistant: ' -}}
+    {%- if (enable_thinking is defined and enable_thinking is false) or enable_thinking is not defined %}
+        {{- '<think>
+
+</think>
+
+' }}
+    {%- endif %}
+    {%- if enable_thinking is defined and enable_thinking is true %}{{- '<think>' }}{%- endif %}
+{%- endif %}

config.json

@@ -0,0 +1,69 @@
+{
+  "architectures": [
+    "Ernie4_5_VLMoeForConditionalGeneration"
+  ],
+  "dtype": "bfloat16",
+  "image_end_token_id": 101305,
+  "image_start_token_id": 101304,
+  "image_token_id": 100295,
+  "model_type": "ernie4_5_vl_moe",
+  "text_config": {
+    "hidden_act": "silu",
+    "hidden_size": 2560,
+    "initializer_range": 0.02,
+    "intermediate_size": 12288,
+    "max_position_embeddings": 131072,
+    "mlp_layer_types": [
+      "dense",
+      "sparse",
+      "sparse",
+      "sparse"
+    ],
+    "model_type": "ernie4_5_vl_moe_text",
+    "moe_intermediate_size": [
+      1536,
+      512
+    ],
+    "moe_k": 6,
+    "moe_norm_min": 1e-12,
+    "moe_num_experts": 64,
+    "moe_num_shared_experts": 2,
+    "num_attention_heads": 20,
+    "num_hidden_layers": 4,
+    "num_key_value_heads": 4,
+    "output_router_logits": false,
+    "rms_norm_eps": 1e-05,
+    "rope_parameters": {
+      "mrope_section": [
+        22,
+        22,
+        20
+      ],
+      "rope_theta": 500000.0,
+      "rope_type": "default"
+    },
+    "router_aux_loss_coef": 0.001,
+    "use_bias": false,
+    "use_cache": true,
+    "vocab_size": 103424
+  },
+  "transformers_version": "5.0.0.dev0",
+  "video_end_token_id": 101307,
+  "video_start_token_id": 101306,
+  "video_token_id": 103367,
+  "vision_config": {
+    "depth": 8,
+    "hidden_act": "quick_gelu",
+    "hidden_size": 1280,
+    "in_channels": 3,
+    "initializer_range": 0.02,
+    "intermediate_size": 5120,
+    "model_type": "ernie4_5_vl_moe_vision",
+    "num_heads": 16,
+    "patch_size": 14,
+    "rms_norm_eps": 1e-06,
+    "spatial_merge_size": 2,
+    "temporal_merge_size": 2
+  },
+  "vocab_size": 509
+}

generation_config.json

@@ -0,0 +1,11 @@
+{
+  "bos_token_id": 1,
+  "do_sample": true,
+  "eos_token_id": 2,
+  "frequency_penalty": 0.0,
+  "pad_token_id": 0,
+  "presence_penalty": 0.0,
+  "temperature": 0.2,
+  "top_p": 0.8,
+  "transformers_version": "5.0.0.dev0"
+}

processor_config.json

@@ -0,0 +1,65 @@
+{
+  "image_processor": {
+    "data_format": "channels_first",
+    "do_convert_rgb": true,
+    "do_normalize": true,
+    "do_rescale": true,
+    "do_resize": true,
+    "image_mean": [
+      0.48145466,
+      0.4578275,
+      0.40821073
+    ],
+    "image_processor_type": "Ernie4_5_VLMoeImageProcessorFast",
+    "image_std": [
+      0.26862954,
+      0.26130258,
+      0.27577711
+    ],
+    "merge_size": 2,
+    "patch_size": 4,
+    "processor_class": "Ernie4_5_VLMoeProcessor",
+    "resample": 3,
+    "rescale_factor": 0.00392156862745098,
+    "size": {
+      "longest_edge": 3136,
+      "shortest_edge": 784
+    }
+  },
+  "processor_class": "Ernie4_5_VLMoeProcessor",
+  "video_processor": {
+    "data_format": "channels_first",
+    "default_to_square": true,
+    "do_convert_rgb": true,
+    "do_normalize": true,
+    "do_rescale": true,
+    "do_resize": true,
+    "do_sample_frames": true,
+    "draw_on_frames": true,
+    "font": "Roboto-Regular.ttf",
+    "image_mean": [
+      0.48145466,
+      0.4578275,
+      0.40821073
+    ],
+    "image_std": [
+      0.26862954,
+      0.26130258,
+      0.27577711
+    ],
+    "max_frames": 180,
+    "merge_size": 2,
+    "min_frames": 16,
+    "patch_size": 4,
+    "processor_class": "Ernie4_5_VLMoeProcessor",
+    "resample": 3,
+    "rescale_factor": 0.00392156862745098,
+    "return_metadata": false,
+    "size": {
+      "longest_edge": 3136,
+      "shortest_edge": 784
+    },
+    "temporal_patch_size": 2,
+    "video_processor_type": "Ernie4_5_VLMoeVideoProcessor"
+  }
+}

tokenizer_config.json

@@ -0,0 +1,50 @@
+{
+  "add_prefix_space": false,
+  "backend": "tokenizers",
+  "bos_token": "<s>",
+  "clean_up_tokenization_spaces": false,
+  "cls_token": "<|begin_of_sentence|>",
+  "eos_token": "</s>",
+  "extra_special_tokens": [
+    "<|IMAGE_PLACEHOLDER|>",
+    "<|IMAGE_START|>",
+    "<|IMAGE_END|>",
+    "<|VIDEO_PLACEHOLDER|>",
+    "<|VIDEO_START|>",
+    "<|VIDEO_END|>",
+    "<think>",
+    "</think>"
+  ],
+  "header_end_token": "<mask:7>",
+  "header_start_token": "<mask:6>",
+  "image_end_token": "<|IMAGE_END|>",
+  "image_start_token": "<|IMAGE_START|>",
+  "image_token": "<|IMAGE_PLACEHOLDER|>",
+  "is_local": true,
+  "mask_token": "<mask:1>",
+  "model_max_length": 131072,
+  "model_specific_special_tokens": {
+    "header_end_token": "<mask:7>",
+    "header_start_token": "<mask:6>",
+    "image_end_token": "<|IMAGE_END|>",
+    "image_start_token": "<|IMAGE_START|>",
+    "image_token": "<|IMAGE_PLACEHOLDER|>",
+    "sys_end_token": "<mask:5>",
+    "sys_start_token": "<mask:4>",
+    "video_end_token": "<|VIDEO_END|>",
+    "video_start_token": "<|VIDEO_START|>",
+    "video_token": "<|VIDEO_PLACEHOLDER|>"
+  },
+  "pad_token": "<unk>",
+  "padding_side": "left",
+  "processor_class": "Ernie4_5_VLMoeProcessor",
+  "sep_token": "<|end_of_sentence|>",
+  "sys_end_token": "<mask:5>",
+  "sys_start_token": "<mask:4>",
+  "tokenizer_class": "LlamaTokenizer",
+  "unk_token": "<unk>",
+  "use_default_system_prompt": false,
+  "video_end_token": "<|VIDEO_END|>",
+  "video_start_token": "<|VIDEO_START|>",
+  "video_token": "<|VIDEO_PLACEHOLDER|>"
+}