Update tiny models for Exaone4_5_VisionModel
Browse files- .gitattributes +1 -0
- chat_template.jinja +161 -0
- config.json +27 -0
- model.safetensors +3 -0
- preprocessor_config.json +28 -0
- tokenizer.json +3 -0
- tokenizer_config.json +21 -0
- video_preprocessor_config.json +32 -0
.gitattributes
CHANGED
|
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
|
| 33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
|
|
|
|
|
| 33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
| 36 |
+
tokenizer.json filter=lfs diff=lfs merge=lfs -text
|
chat_template.jinja
ADDED
|
@@ -0,0 +1,161 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{% set image_count = namespace(value=0) %}
|
| 2 |
+
{% set video_count = namespace(value=0) %}
|
| 3 |
+
|
| 4 |
+
{%- set role_indicators = {
|
| 5 |
+
'user': '<|user|>\n',
|
| 6 |
+
'assistant': '<|assistant|>\n',
|
| 7 |
+
'system': '<|system|>\n',
|
| 8 |
+
'tool': '<|tool|>\n',
|
| 9 |
+
'tool_declare': '<|tool_declare|>\n'
|
| 10 |
+
} %}
|
| 11 |
+
{%- set end_of_turn = '<|endofturn|>\n' %}
|
| 12 |
+
|
| 13 |
+
|
| 14 |
+
{%- macro declare_available_tools(tools) %}
|
| 15 |
+
{{- "# Tools" }}
|
| 16 |
+
{{- "\n" }}
|
| 17 |
+
{%- for tool in tools %}
|
| 18 |
+
{{- "<tool>" }}
|
| 19 |
+
{{- tool | tojson(ensure_ascii=False) | safe }}
|
| 20 |
+
{{- "</tool>\n" }}
|
| 21 |
+
{%- endfor %}
|
| 22 |
+
{%- endmacro %}
|
| 23 |
+
|
| 24 |
+
|
| 25 |
+
{%- set ns = namespace(last_query_index = messages|length - 1, last_query_index_not_yet_determined = true) %}
|
| 26 |
+
{%- for message in messages[::-1] %}
|
| 27 |
+
{%- set index = (messages|length - 1) - loop.index0 %}
|
| 28 |
+
{%- if ns.last_query_index_not_yet_determined and message.role == "user" and message.content is string %}
|
| 29 |
+
{%- set ns.last_query_index = index -%}
|
| 30 |
+
{%- set ns.last_query_index_not_yet_determined = false -%}
|
| 31 |
+
{%- endif %}
|
| 32 |
+
{%- endfor %}
|
| 33 |
+
|
| 34 |
+
{%- if tools is defined and tools %}
|
| 35 |
+
{{- role_indicators['tool_declare'] }}
|
| 36 |
+
{{- declare_available_tools(tools) }}
|
| 37 |
+
{{- end_of_turn -}}
|
| 38 |
+
{%- endif %}
|
| 39 |
+
|
| 40 |
+
{%- for i in range(messages | length) %}
|
| 41 |
+
{%- set msg = messages[i] %}
|
| 42 |
+
{%- set role = msg.role %}
|
| 43 |
+
{%- if role not in role_indicators %}
|
| 44 |
+
{{- raise_exception('Unknown role: ' ~ role) }}
|
| 45 |
+
{%- endif %}
|
| 46 |
+
|
| 47 |
+
{%- if i == 0 %}
|
| 48 |
+
{%- if role == 'system' %}
|
| 49 |
+
{{- role_indicators['system'] }}
|
| 50 |
+
{{- msg.content }}
|
| 51 |
+
{{- end_of_turn -}}
|
| 52 |
+
{%- continue %}
|
| 53 |
+
{%- endif %}
|
| 54 |
+
{%- endif %}
|
| 55 |
+
|
| 56 |
+
{%- if role == 'assistant' %}
|
| 57 |
+
{{- role_indicators['assistant'] }}
|
| 58 |
+
|
| 59 |
+
{%- set content = (msg.content if (msg.content is defined and msg.content) else "") -%}
|
| 60 |
+
{%- set reasoning = none -%}
|
| 61 |
+
|
| 62 |
+
{%- if msg.reasoning_content is defined and msg.reasoning_content%}
|
| 63 |
+
{%- set reasoning = msg.reasoning_content.strip() -%}
|
| 64 |
+
{%- elif content and "</think>" in content %}
|
| 65 |
+
{%- set _parts = content.split('</think>') -%}
|
| 66 |
+
{%- set reasoning = _parts[0].lstrip('<think>').strip() -%}
|
| 67 |
+
{%- set content = _parts[-1].strip() -%}
|
| 68 |
+
{%- endif %}
|
| 69 |
+
|
| 70 |
+
{%- if not (reasoning and i > ns.last_query_index) or (skip_think is defined and skip_think) %}
|
| 71 |
+
{%- set reasoning = none %}
|
| 72 |
+
{%- endif %}
|
| 73 |
+
|
| 74 |
+
{%- set content = content.strip() -%}
|
| 75 |
+
|
| 76 |
+
{{- "<think>\n" }}
|
| 77 |
+
{{- (reasoning if reasoning is not none else "") }}
|
| 78 |
+
{{- "\n</think>\n\n" }}
|
| 79 |
+
|
| 80 |
+
{{- content }}
|
| 81 |
+
|
| 82 |
+
{%- if msg.tool_calls %}
|
| 83 |
+
{%- if content is defined and content %}
|
| 84 |
+
{{- "\n" }}
|
| 85 |
+
{%- endif %}
|
| 86 |
+
{%- for tool_call in msg.tool_calls %}
|
| 87 |
+
{%- if tool_call.function is defined %}
|
| 88 |
+
{%- set tool_call = tool_call.function %}
|
| 89 |
+
{%- endif %}
|
| 90 |
+
|
| 91 |
+
{%- if tool_call.arguments is defined %}
|
| 92 |
+
{%- set arguments = tool_call.arguments %}
|
| 93 |
+
{%- elif tool_call.parameters is defined %}
|
| 94 |
+
{%- set arguments = tool_call.parameters %}
|
| 95 |
+
{%- else %}
|
| 96 |
+
{{- raise_exception('arguments or parameters are mandatory: ' ~ tool_call) }}
|
| 97 |
+
{%- endif %}
|
| 98 |
+
{%- if arguments is string %}
|
| 99 |
+
{{- "<tool_call>" }}{"name": "{{- tool_call.name }}", "arguments": {{ arguments }}}{{- "</tool_call>" }}
|
| 100 |
+
{%- else %}
|
| 101 |
+
{{- "<tool_call>" }}{"name": "{{- tool_call.name }}", "arguments": {{ arguments | tojson(ensure_ascii=False) | safe }}}{{- "</tool_call>" }}
|
| 102 |
+
{%- endif %}
|
| 103 |
+
{%- if not loop.last %}
|
| 104 |
+
{{- "\n" }}
|
| 105 |
+
{%- endif %}
|
| 106 |
+
|
| 107 |
+
{%- endfor %}
|
| 108 |
+
{%- endif %}
|
| 109 |
+
{{- end_of_turn -}}
|
| 110 |
+
|
| 111 |
+
{%- elif role == "tool" %}
|
| 112 |
+
{%- if i == 0 or messages[i - 1].role != "tool" %}
|
| 113 |
+
{{- role_indicators['tool'] }}
|
| 114 |
+
{%- endif %}
|
| 115 |
+
{%- if msg.content is defined %}
|
| 116 |
+
{%- if msg.content is string %}
|
| 117 |
+
{{- "<tool_result>" }}{{ msg.content }}{{- "</tool_result>" }}
|
| 118 |
+
{%- else %}
|
| 119 |
+
{{- "<tool_result>" }}{{ msg.content | tojson(ensure_ascii=False) | safe }}{{- "</tool_result>" }}
|
| 120 |
+
{%- endif %}
|
| 121 |
+
{%- endif %}
|
| 122 |
+
{%- if loop.last or messages[i + 1].role != "tool" %}
|
| 123 |
+
{{- end_of_turn -}}
|
| 124 |
+
{%- else %}
|
| 125 |
+
{{- "\n" }}
|
| 126 |
+
{%- endif %}
|
| 127 |
+
|
| 128 |
+
{%- else %}
|
| 129 |
+
{{- role_indicators[role] }}
|
| 130 |
+
{%- if msg.content is string %}
|
| 131 |
+
{{- msg.content }}
|
| 132 |
+
{%- else %}
|
| 133 |
+
{%- for content in msg.content %}
|
| 134 |
+
{%- if content.type == 'image' %}
|
| 135 |
+
{%- set image_count.value = image_count.value + 1 %}
|
| 136 |
+
{%- if add_vision_id %}Picture {{ image_count.value }}: {% endif %}
|
| 137 |
+
{{- "<vision><|image_pad|></vision>\n" }}
|
| 138 |
+
{%- elif content.type == 'video' %}
|
| 139 |
+
{%- set video_count.value = video_count.value + 1 %}
|
| 140 |
+
{%- if add_vision_id %}Video {{ video_count.value }}: {% endif %}
|
| 141 |
+
{{- "<vision><|video_pad|></vision>\n" }}
|
| 142 |
+
{%- elif content.type == 'text' %}
|
| 143 |
+
{{- content.text }}
|
| 144 |
+
{%- else %}
|
| 145 |
+
{{- content.text }}
|
| 146 |
+
{%- endif %}
|
| 147 |
+
{%- endfor %}
|
| 148 |
+
{%- endif %}
|
| 149 |
+
{{- end_of_turn -}}
|
| 150 |
+
{%- endif %}
|
| 151 |
+
{% endfor %}
|
| 152 |
+
|
| 153 |
+
|
| 154 |
+
{%- if add_generation_prompt %}
|
| 155 |
+
{{- role_indicators['assistant'] }}
|
| 156 |
+
{%- if enable_thinking is not defined or enable_thinking is true %}
|
| 157 |
+
{{- "<think>\n" }}
|
| 158 |
+
{%- else %}
|
| 159 |
+
{{- "<think>\n\n</think>\n\n" }}
|
| 160 |
+
{%- endif %}
|
| 161 |
+
{%- endif %}
|
config.json
ADDED
|
@@ -0,0 +1,27 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"architectures": [
|
| 3 |
+
"Exaone4_5_VisionModel"
|
| 4 |
+
],
|
| 5 |
+
"depth": 2,
|
| 6 |
+
"dtype": "float32",
|
| 7 |
+
"fullatt_block_indexes": [
|
| 8 |
+
7,
|
| 9 |
+
15,
|
| 10 |
+
23,
|
| 11 |
+
31
|
| 12 |
+
],
|
| 13 |
+
"hidden_act": "silu",
|
| 14 |
+
"hidden_size": 32,
|
| 15 |
+
"in_channels": 3,
|
| 16 |
+
"initializer_range": 0.02,
|
| 17 |
+
"intermediate_size": 32,
|
| 18 |
+
"model_type": "exaone4_5_vision",
|
| 19 |
+
"num_heads": 4,
|
| 20 |
+
"num_key_value_heads": 2,
|
| 21 |
+
"out_hidden_size": 32,
|
| 22 |
+
"patch_size": 16,
|
| 23 |
+
"spatial_merge_size": 1,
|
| 24 |
+
"temporal_patch_size": 2,
|
| 25 |
+
"transformers_version": "5.16.0.dev0",
|
| 26 |
+
"window_size": 112
|
| 27 |
+
}
|
model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:3f398f8598c88aba70ba1f0df2c86da88216c21d4720659a1278151c93fafb70
|
| 3 |
+
size 259008
|
preprocessor_config.json
ADDED
|
@@ -0,0 +1,28 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"data_format": "channels_first",
|
| 3 |
+
"default_to_square": true,
|
| 4 |
+
"do_convert_rgb": true,
|
| 5 |
+
"do_normalize": true,
|
| 6 |
+
"do_rescale": true,
|
| 7 |
+
"do_resize": true,
|
| 8 |
+
"image_mean": [
|
| 9 |
+
0.48145466,
|
| 10 |
+
0.4578275,
|
| 11 |
+
0.40821073
|
| 12 |
+
],
|
| 13 |
+
"image_processor_type": "Qwen2VLImageProcessor",
|
| 14 |
+
"image_std": [
|
| 15 |
+
0.26862954,
|
| 16 |
+
0.26130258,
|
| 17 |
+
0.27577711
|
| 18 |
+
],
|
| 19 |
+
"merge_size": 2,
|
| 20 |
+
"patch_size": 14,
|
| 21 |
+
"resample": 3,
|
| 22 |
+
"rescale_factor": 0.00392156862745098,
|
| 23 |
+
"size": {
|
| 24 |
+
"longest_edge": 3211264,
|
| 25 |
+
"shortest_edge": 3136
|
| 26 |
+
},
|
| 27 |
+
"temporal_patch_size": 2
|
| 28 |
+
}
|
tokenizer.json
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:0bd798efa30739e209d51f36cfc2f0a636711e37ad9d69b77e4e5c8ca5f09fab
|
| 3 |
+
size 12160205
|
tokenizer_config.json
ADDED
|
@@ -0,0 +1,21 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"backend": "tokenizers",
|
| 3 |
+
"bos_token": "[BOS]",
|
| 4 |
+
"clean_up_tokenization_spaces": false,
|
| 5 |
+
"eos_token": "<|endofturn|>",
|
| 6 |
+
"errors": "replace",
|
| 7 |
+
"is_local": true,
|
| 8 |
+
"local_files_only": false,
|
| 9 |
+
"model_input_names": [
|
| 10 |
+
"input_ids",
|
| 11 |
+
"attention_mask"
|
| 12 |
+
],
|
| 13 |
+
"model_max_length": 1000000000000000019884624838656,
|
| 14 |
+
"model_specific_special_tokens": {},
|
| 15 |
+
"pad_token": "[PAD]",
|
| 16 |
+
"padding_side": "right",
|
| 17 |
+
"processor_class": "Exaone4_5_Processor",
|
| 18 |
+
"split_special_tokens": false,
|
| 19 |
+
"tokenizer_class": "TokenizersBackend",
|
| 20 |
+
"unk_token": "[UNK]"
|
| 21 |
+
}
|
video_preprocessor_config.json
ADDED
|
@@ -0,0 +1,32 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"data_format": "channels_first",
|
| 3 |
+
"default_to_square": true,
|
| 4 |
+
"do_convert_rgb": true,
|
| 5 |
+
"do_normalize": true,
|
| 6 |
+
"do_rescale": true,
|
| 7 |
+
"do_resize": true,
|
| 8 |
+
"do_sample_frames": false,
|
| 9 |
+
"image_mean": [
|
| 10 |
+
0.48145466,
|
| 11 |
+
0.4578275,
|
| 12 |
+
0.40821073
|
| 13 |
+
],
|
| 14 |
+
"image_std": [
|
| 15 |
+
0.26862954,
|
| 16 |
+
0.26130258,
|
| 17 |
+
0.27577711
|
| 18 |
+
],
|
| 19 |
+
"max_frames": 768,
|
| 20 |
+
"merge_size": 2,
|
| 21 |
+
"min_frames": 4,
|
| 22 |
+
"patch_size": 14,
|
| 23 |
+
"resample": 3,
|
| 24 |
+
"rescale_factor": 0.00392156862745098,
|
| 25 |
+
"return_metadata": false,
|
| 26 |
+
"size": {
|
| 27 |
+
"longest_edge": 3211264,
|
| 28 |
+
"shortest_edge": 3136
|
| 29 |
+
},
|
| 30 |
+
"temporal_patch_size": 2,
|
| 31 |
+
"video_processor_type": "Qwen2VLVideoProcessor"
|
| 32 |
+
}
|