Image-Text-to-Text
Transformers
Safetensors
minimax_m3_vl
multimodal
Mixture of Experts
agent
coding
video
conversational
custom_code
8-bit precision
Instructions to use sparkarena/Minimax-M3-v0-NVFP4 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use sparkarena/Minimax-M3-v0-NVFP4 with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-text-to-text", model="sparkarena/Minimax-M3-v0-NVFP4", trust_remote_code=True) messages = [ { "role": "user", "content": [ {"type": "image", "url": "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/p-blog/candy.JPG"}, {"type": "text", "text": "What animal is on the candy?"} ] }, ] pipe(text=messages)# Load model directly from transformers import AutoProcessor, AutoModelForMultimodalLM processor = AutoProcessor.from_pretrained("sparkarena/Minimax-M3-v0-NVFP4", trust_remote_code=True) model = AutoModelForMultimodalLM.from_pretrained("sparkarena/Minimax-M3-v0-NVFP4", trust_remote_code=True) messages = [ { "role": "user", "content": [ {"type": "image", "url": "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/p-blog/candy.JPG"}, {"type": "text", "text": "What animal is on the candy?"} ] }, ] inputs = processor.apply_chat_template( messages, add_generation_prompt=True, tokenize=True, return_dict=True, return_tensors="pt", ).to(model.device) outputs = model.generate(**inputs, max_new_tokens=40) print(processor.decode(outputs[0][inputs["input_ids"].shape[-1]:])) - Notebooks
- Google Colab
- Kaggle
- Local Apps Settings
- vLLM
How to use sparkarena/Minimax-M3-v0-NVFP4 with vLLM:
Install from pip and serve model
# Install vLLM from pip: pip install vllm # Start the vLLM server: vllm serve "sparkarena/Minimax-M3-v0-NVFP4" # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:8000/v1/chat/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "sparkarena/Minimax-M3-v0-NVFP4", "messages": [ { "role": "user", "content": [ { "type": "text", "text": "Describe this image in one sentence." }, { "type": "image_url", "image_url": { "url": "https://cdn.britannica.com/61/93061-050-99147DCE/Statue-of-Liberty-Island-New-York-Bay.jpg" } } ] } ] }'Use Docker
docker model run hf.co/sparkarena/Minimax-M3-v0-NVFP4
- SGLang
How to use sparkarena/Minimax-M3-v0-NVFP4 with SGLang:
Install from pip and serve model
# Install SGLang from pip: pip install sglang # Start the SGLang server: python3 -m sglang.launch_server \ --model-path "sparkarena/Minimax-M3-v0-NVFP4" \ --host 0.0.0.0 \ --port 30000 # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:30000/v1/chat/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "sparkarena/Minimax-M3-v0-NVFP4", "messages": [ { "role": "user", "content": [ { "type": "text", "text": "Describe this image in one sentence." }, { "type": "image_url", "image_url": { "url": "https://cdn.britannica.com/61/93061-050-99147DCE/Statue-of-Liberty-Island-New-York-Bay.jpg" } } ] } ] }'Use Docker images
docker run --gpus all \ --shm-size 32g \ -p 30000:30000 \ -v ~/.cache/huggingface:/root/.cache/huggingface \ --env "HF_TOKEN=<secret>" \ --ipc=host \ lmsysorg/sglang:latest \ python3 -m sglang.launch_server \ --model-path "sparkarena/Minimax-M3-v0-NVFP4" \ --host 0.0.0.0 \ --port 30000 # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:30000/v1/chat/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "sparkarena/Minimax-M3-v0-NVFP4", "messages": [ { "role": "user", "content": [ { "type": "text", "text": "Describe this image in one sentence." }, { "type": "image_url", "image_url": { "url": "https://cdn.britannica.com/61/93061-050-99147DCE/Statue-of-Liberty-Island-New-York-Bay.jpg" } } ] } ] }' - Docker Model Runner
How to use sparkarena/Minimax-M3-v0-NVFP4 with Docker Model Runner:
docker model run hf.co/sparkarena/Minimax-M3-v0-NVFP4
Add files using upload-large-folder tool
Browse files- .gitattributes +7 -32
- README.md +11 -0
- added_tokens.json +63 -0
- chat_template.jinja +256 -0
- config.json +305 -0
- configuration_minimax_m3_vl.py +111 -0
- generation_config.json +13 -0
- image_processor.py +223 -0
- merges.txt +0 -0
- model-00001-of-00046.safetensors +3 -0
- model-00003-of-00046.safetensors +3 -0
- model-00004-of-00046.safetensors +3 -0
- model-00005-of-00046.safetensors +3 -0
- model-00006-of-00046.safetensors +3 -0
- model-00007-of-00046.safetensors +3 -0
- model-00008-of-00046.safetensors +3 -0
- model-00010-of-00046.safetensors +3 -0
- model-00011-of-00046.safetensors +3 -0
- model-00012-of-00046.safetensors +3 -0
- model-00016-of-00046.safetensors +3 -0
- model-00017-of-00046.safetensors +3 -0
- model-00019-of-00046.safetensors +3 -0
- model-00021-of-00046.safetensors +3 -0
- model-00022-of-00046.safetensors +3 -0
- model-00023-of-00046.safetensors +3 -0
- model-00024-of-00046.safetensors +3 -0
- model-00025-of-00046.safetensors +3 -0
- model-00027-of-00046.safetensors +3 -0
- model-00029-of-00046.safetensors +3 -0
- model-00030-of-00046.safetensors +3 -0
- model-00032-of-00046.safetensors +3 -0
- model-00033-of-00046.safetensors +3 -0
- model-00034-of-00046.safetensors +3 -0
- model-00035-of-00046.safetensors +3 -0
- model-00038-of-00046.safetensors +3 -0
- model-00041-of-00046.safetensors +3 -0
- model-00042-of-00046.safetensors +3 -0
- model-00043-of-00046.safetensors +3 -0
- model-00044-of-00046.safetensors +3 -0
- model-00045-of-00046.safetensors +3 -0
- model.safetensors.index.json +0 -0
- preprocessor_config.json +32 -0
- processing_minimax.py +254 -0
- special_tokens_map.json +16 -0
- tokenizer.json +0 -0
- tokenizer_config.json +502 -0
- video_processor.py +208 -0
- vocab.json +0 -0
.gitattributes
CHANGED
|
@@ -1,35 +1,10 @@
|
|
| 1 |
-
*.
|
| 2 |
-
*.arrow filter=lfs diff=lfs merge=lfs -text
|
| 3 |
*.bin filter=lfs diff=lfs merge=lfs -text
|
| 4 |
-
*.bz2 filter=lfs diff=lfs merge=lfs -text
|
| 5 |
-
*.ckpt filter=lfs diff=lfs merge=lfs -text
|
| 6 |
-
*.ftz filter=lfs diff=lfs merge=lfs -text
|
| 7 |
-
*.gz filter=lfs diff=lfs merge=lfs -text
|
| 8 |
-
*.h5 filter=lfs diff=lfs merge=lfs -text
|
| 9 |
-
*.joblib filter=lfs diff=lfs merge=lfs -text
|
| 10 |
-
*.lfs.* filter=lfs diff=lfs merge=lfs -text
|
| 11 |
-
*.mlmodel filter=lfs diff=lfs merge=lfs -text
|
| 12 |
-
*.model filter=lfs diff=lfs merge=lfs -text
|
| 13 |
-
*.msgpack filter=lfs diff=lfs merge=lfs -text
|
| 14 |
-
*.npy filter=lfs diff=lfs merge=lfs -text
|
| 15 |
-
*.npz filter=lfs diff=lfs merge=lfs -text
|
| 16 |
-
*.onnx filter=lfs diff=lfs merge=lfs -text
|
| 17 |
-
*.ot filter=lfs diff=lfs merge=lfs -text
|
| 18 |
-
*.parquet filter=lfs diff=lfs merge=lfs -text
|
| 19 |
-
*.pb filter=lfs diff=lfs merge=lfs -text
|
| 20 |
-
*.pickle filter=lfs diff=lfs merge=lfs -text
|
| 21 |
-
*.pkl filter=lfs diff=lfs merge=lfs -text
|
| 22 |
*.pt filter=lfs diff=lfs merge=lfs -text
|
| 23 |
*.pth filter=lfs diff=lfs merge=lfs -text
|
| 24 |
-
*.
|
| 25 |
-
*.
|
| 26 |
-
|
| 27 |
-
*.
|
| 28 |
-
*.
|
| 29 |
-
*.
|
| 30 |
-
*.tgz filter=lfs diff=lfs merge=lfs -text
|
| 31 |
-
*.wasm filter=lfs diff=lfs merge=lfs -text
|
| 32 |
-
*.xz 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
|
|
|
|
| 1 |
+
*.safetensors filter=lfs diff=lfs merge=lfs -text
|
|
|
|
| 2 |
*.bin filter=lfs diff=lfs merge=lfs -text
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 3 |
*.pt filter=lfs diff=lfs merge=lfs -text
|
| 4 |
*.pth filter=lfs diff=lfs merge=lfs -text
|
| 5 |
+
*.onnx filter=lfs diff=lfs merge=lfs -text
|
| 6 |
+
*.json text
|
| 7 |
+
*.py text
|
| 8 |
+
*.jinja text
|
| 9 |
+
*.txt text
|
| 10 |
+
*.md text
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
README.md
CHANGED
|
@@ -23,6 +23,17 @@ use the calibration from this model.
|
|
| 23 |
|
| 24 |
The calibration is still a work in progress (hence "v0" model). Updates are planned to improve performance.
|
| 25 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 26 |
<div align="center">
|
| 27 |
<img width="60%" src="figures/logo.svg" alt="MiniMax">
|
| 28 |
</div>
|
|
|
|
| 23 |
|
| 24 |
The calibration is still a work in progress (hence "v0" model). Updates are planned to improve performance.
|
| 25 |
|
| 26 |
+
Run with sparkrun; part of Spark Arena
|
| 27 |
+
|
| 28 |
+
https://sparkrun.dev
|
| 29 |
+
https://spark-arena.com
|
| 30 |
+
|
| 31 |
+
To run with sparkrun on 4x DGX Spark Nodes:
|
| 32 |
+
```
|
| 33 |
+
sparkrun run @experimental/minimax-m3-v0-nvfp4-4x
|
| 34 |
+
```
|
| 35 |
+
|
| 36 |
+
|
| 37 |
<div align="center">
|
| 38 |
<img width="60%" src="figures/logo.svg" alt="MiniMax">
|
| 39 |
</div>
|
added_tokens.json
ADDED
|
@@ -0,0 +1,63 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"]!p~[": 200000,
|
| 3 |
+
"<fim_prefix>": 200001,
|
| 4 |
+
"<fim_middle>": 200002,
|
| 5 |
+
"<fim_suffix>": 200003,
|
| 6 |
+
"<fim_pad>": 200004,
|
| 7 |
+
"<reponame>": 200005,
|
| 8 |
+
"<filename>": 200006,
|
| 9 |
+
"<gh_stars>": 200007,
|
| 10 |
+
"<issue_start>": 200008,
|
| 11 |
+
"<issue_comment>": 200009,
|
| 12 |
+
"<issue_closed>": 200010,
|
| 13 |
+
"<jupyter_start>": 200011,
|
| 14 |
+
"<jupyter_text>": 200012,
|
| 15 |
+
"<jupyter_code>": 200013,
|
| 16 |
+
"<jupyter_output>": 200014,
|
| 17 |
+
"<empty_output>": 200015,
|
| 18 |
+
"<commit_before>": 200016,
|
| 19 |
+
"<commit_msg>": 200017,
|
| 20 |
+
"<commit_after>": 200018,
|
| 21 |
+
"]~b]": 200019,
|
| 22 |
+
"[e~[": 200020,
|
| 23 |
+
"]!d~[": 200021,
|
| 24 |
+
"<function_call>": 200022,
|
| 25 |
+
"<code_interpreter>": 200023,
|
| 26 |
+
"]<]speech[>[": 200024,
|
| 27 |
+
"]<]image[>[": 200025,
|
| 28 |
+
"]<]video[>[": 200026,
|
| 29 |
+
"]<]start of speech[>[": 200027,
|
| 30 |
+
"]<]end of speech[>[": 200028,
|
| 31 |
+
"]<]start of image[>[": 200029,
|
| 32 |
+
"]<]end of image[>[": 200030,
|
| 33 |
+
"]<]start of video[>[": 200031,
|
| 34 |
+
"]<]end of video[>[": 200032,
|
| 35 |
+
"]<]vision pad[>[": 200033,
|
| 36 |
+
"]~!b[": 200034,
|
| 37 |
+
"<jupyter_error>": 200035,
|
| 38 |
+
"<add_file>": 200036,
|
| 39 |
+
"<delete_file>": 200037,
|
| 40 |
+
"<rename_file>": 200038,
|
| 41 |
+
"<edit_file>": 200039,
|
| 42 |
+
"<commit_message>": 200040,
|
| 43 |
+
"<empty_source_file>": 200041,
|
| 44 |
+
"<repo_struct>": 200042,
|
| 45 |
+
"<code_context>": 200043,
|
| 46 |
+
"<file_content>": 200044,
|
| 47 |
+
"<source_files>": 200045,
|
| 48 |
+
"<pr_start>": 200046,
|
| 49 |
+
"<review_comment>": 200047,
|
| 50 |
+
"<filepath>": 200048,
|
| 51 |
+
"<file_sep>": 200049,
|
| 52 |
+
"<think>": 200050,
|
| 53 |
+
"</think>": 200051,
|
| 54 |
+
"<tool_call>": 200052,
|
| 55 |
+
"</tool_call>": 200053,
|
| 56 |
+
"]<]frame[>[": 200054,
|
| 57 |
+
"]<]start of frame[>[": 200055,
|
| 58 |
+
"]<]end of frame[>[": 200056,
|
| 59 |
+
"<|content_altered_placeholder|>": 200057,
|
| 60 |
+
"]<]minimax[>[": 200058,
|
| 61 |
+
"<mm:think>": 200059,
|
| 62 |
+
"</mm:think>": 200060
|
| 63 |
+
}
|
chat_template.jinja
ADDED
|
@@ -0,0 +1,256 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{# ---------- special token variables ---------- #}
|
| 2 |
+
{%- set ns_token = ']<]minimax[>[' -%}
|
| 3 |
+
{%- set bod_token = ']~!b[' -%}
|
| 4 |
+
{%- set bos_token = ']~b]' -%}
|
| 5 |
+
{%- set eos_token = '[e~[' -%}
|
| 6 |
+
{%- set toolcall_begin_token = ns_token ~ '<tool_call>' -%}
|
| 7 |
+
{%- set toolcall_end_token = ns_token ~ '</tool_call>' -%}
|
| 8 |
+
{%- set think_begin_token = '<mm:think>' -%}
|
| 9 |
+
{%- set think_end_token = '</mm:think>' -%}
|
| 10 |
+
{%- set image_token = ']<]image[>[' -%}
|
| 11 |
+
{%- set video_token = ']<]video[>[' -%}
|
| 12 |
+
{#- Thinking mode: "enabled" / "disabled" / "adaptive" / not defined -#}
|
| 13 |
+
{#- Recursive XML renderer for tool_call arguments ======================== -#}
|
| 14 |
+
{#- None values are intentionally skipped in mapping iteration so that
|
| 15 |
+
`<key>null</key>` (which would round-trip to the literal string "null")
|
| 16 |
+
never appears in the rendered tool_call. The convention is: omit the
|
| 17 |
+
field entirely. The top-level `_args` loop applies the same rule.
|
| 18 |
+
The `val is none` branch below is a safety net only — upstream cleaning
|
| 19 |
+
(drop_none_in_tool_arguments) should ensure no None ever reaches here. -#}
|
| 20 |
+
{%- macro to_xml(val, ns) -%}
|
| 21 |
+
{%- if val is mapping -%}
|
| 22 |
+
{%- for k, v in val.items() if v is not none -%}
|
| 23 |
+
{{ ns }}<{{ k }}>{{ to_xml(v, ns) }}{{ ns }}</{{ k }}>
|
| 24 |
+
{%- endfor -%}
|
| 25 |
+
{%- elif val is iterable and val is not string -%}
|
| 26 |
+
{%- for item in val -%}
|
| 27 |
+
{{ ns }}<item>{{ to_xml(item, ns) }}{{ ns }}</item>
|
| 28 |
+
{%- endfor -%}
|
| 29 |
+
{%- elif val is none -%}
|
| 30 |
+
{#- Should be unreachable when upstream cleaning is applied. -#}
|
| 31 |
+
{%- elif val is boolean -%}
|
| 32 |
+
{{ val | tojson }}
|
| 33 |
+
{%- else -%}
|
| 34 |
+
{{ val }}
|
| 35 |
+
{%- endif -%}
|
| 36 |
+
{%- endmacro -%}
|
| 37 |
+
{#- Tool Rendering Functions ============================================== -#}
|
| 38 |
+
{%- macro render_tool_namespace(namespace_name, tool_list) -%}
|
| 39 |
+
{%- for tool in tool_list -%}
|
| 40 |
+
<tool>{{ tool.function | tojson(ensure_ascii=False) }}</tool>
|
| 41 |
+
{% endfor -%}
|
| 42 |
+
{%- endmacro -%}
|
| 43 |
+
{%- macro visible_text(content) -%}
|
| 44 |
+
{%- if content is string -%}
|
| 45 |
+
{{ content }}
|
| 46 |
+
{%- elif content is iterable and content is not mapping -%}
|
| 47 |
+
{%- for item in content -%}
|
| 48 |
+
{%- if item is mapping and item.type == 'text' -%}
|
| 49 |
+
{{- item.text }}
|
| 50 |
+
{%- elif item is mapping and item.type == 'image' -%}
|
| 51 |
+
{{- image_token }}
|
| 52 |
+
{%- elif item is mapping and item.type == 'video' -%}
|
| 53 |
+
{{- video_token}}
|
| 54 |
+
{%- elif item is string -%}
|
| 55 |
+
{{- item }}
|
| 56 |
+
{%- endif -%}
|
| 57 |
+
{%- endfor -%}
|
| 58 |
+
{%- elif content is none -%}
|
| 59 |
+
{{- '' }}
|
| 60 |
+
{%- else -%}
|
| 61 |
+
{{- content }}
|
| 62 |
+
{%- endif -%}
|
| 63 |
+
{%- endmacro -%}
|
| 64 |
+
{#- System Message Construction ============================================ -#}
|
| 65 |
+
{%- macro build_system_message(system_message) -%}
|
| 66 |
+
{%- if system_message and system_message.content -%}
|
| 67 |
+
{{- visible_text(system_message.content) }}
|
| 68 |
+
{#- Handle current_date -#}
|
| 69 |
+
{%- if system_message.current_date -%}
|
| 70 |
+
{{- '\n' ~ 'Current date: ' + system_message.current_date }}
|
| 71 |
+
{%- endif -%}
|
| 72 |
+
{#- Handle current_location -#}
|
| 73 |
+
{%- if system_message.current_location -%}
|
| 74 |
+
{{- '\n' ~ 'Current location: ' + system_message.current_location }}
|
| 75 |
+
{%- endif -%}
|
| 76 |
+
{%- endif -%}
|
| 77 |
+
|
| 78 |
+
{#- Thinking mode instructions -#}
|
| 79 |
+
{{- '\n\n<thinking_instructions>\n' }}
|
| 80 |
+
{{- 'You have a thinking capability that allows you to reason step by step before responding. When thinking is enabled, wrap your reasoning in ' ~ think_begin_token ~ think_end_token ~ ' tags before your response. When thinking is disabled, begin your response directly after the ' ~ think_end_token ~ ' prefix. When thinking is adaptive, decide on your own whether to think for the current turn.\n' }}
|
| 81 |
+
{%- if thinking_mode is defined -%}
|
| 82 |
+
{%- if thinking_mode == "enabled" -%}
|
| 83 |
+
{{- 'Current thinking mode: enabled. You MUST think step by step before every response, including after receiving function/tool results.\n' }}
|
| 84 |
+
{%- elif thinking_mode == "disabled" -%}
|
| 85 |
+
{{- 'Current thinking mode: disabled. Do not output any thinking process.\n' }}
|
| 86 |
+
{%- elif thinking_mode == "adaptive" -%}
|
| 87 |
+
{{- 'Current thinking mode: adaptive. You are encouraged to think for complex decision-making, multi-step reasoning, or when analyzing function/tool results.\n' }}
|
| 88 |
+
{%- endif -%}
|
| 89 |
+
{%- else -%}
|
| 90 |
+
{{- 'Current thinking mode: adaptive. You are encouraged to think for complex decision-making, multi-step reasoning, or when analyzing function/tool results.\n' }}
|
| 91 |
+
{%- endif -%}
|
| 92 |
+
{{- '</thinking_instructions>' }}
|
| 93 |
+
{%- if template_model_id is defined -%}
|
| 94 |
+
{{- '\n\n<model_id>' ~ template_model_id ~ '</model_id>' }}
|
| 95 |
+
{%- endif -%}
|
| 96 |
+
{%- endmacro -%}
|
| 97 |
+
{%- macro build_developer_message(developer_message) -%}
|
| 98 |
+
{%- if developer_message and developer_message.content -%}
|
| 99 |
+
{{- visible_text(developer_message.content) }}
|
| 100 |
+
{%- else -%}
|
| 101 |
+
{%- if model_identity is not defined -%}
|
| 102 |
+
{%- set model_identity = "You are a helpful assistant." -%}
|
| 103 |
+
{%- endif -%}
|
| 104 |
+
{{- model_identity }}
|
| 105 |
+
{%- endif -%}
|
| 106 |
+
{%- endmacro -%}
|
| 107 |
+
{#- Main Template Logic ================================================= -#}
|
| 108 |
+
{#- Role mapping: root -> system sp (high priority), system/developer -> developer sp (low priority) -#}
|
| 109 |
+
{%- set system_message = none -%}
|
| 110 |
+
{%- set developer_message = none -%}
|
| 111 |
+
{%- set conversation_messages = messages -%}
|
| 112 |
+
{%- if messages and messages[0].role == "root" -%}
|
| 113 |
+
{%- set system_message = messages[0] -%}
|
| 114 |
+
{%- set conversation_messages = messages[1:] -%}
|
| 115 |
+
{%- if conversation_messages and conversation_messages[0].role in ["system", "developer"] -%}
|
| 116 |
+
{%- set developer_message = conversation_messages[0] -%}
|
| 117 |
+
{%- set conversation_messages = conversation_messages[1:] -%}
|
| 118 |
+
{%- endif -%}
|
| 119 |
+
{%- elif messages and messages[0].role in ["system", "developer"] -%}
|
| 120 |
+
{%- set developer_message = messages[0] -%}
|
| 121 |
+
{%- set conversation_messages = messages[1:] -%}
|
| 122 |
+
{%- endif -%}
|
| 123 |
+
{#- Render system sp (higher priority, root role only) -#}
|
| 124 |
+
{{- bod_token ~ bos_token ~ 'system' ~ '\n' }}
|
| 125 |
+
{{- build_system_message(system_message) }}
|
| 126 |
+
{{- eos_token ~ '\n' }}
|
| 127 |
+
|
| 128 |
+
{#- Render developer sp (lower priority: system/developer role + tools) -#}
|
| 129 |
+
{{- bos_token ~ 'developer' ~ '\n' }}
|
| 130 |
+
{{- build_developer_message(developer_message) }}
|
| 131 |
+
{%- if tools -%}
|
| 132 |
+
{{- '\n\n' ~ '# Tools' ~ '\n' ~ 'You may call one or more tools to assist with the user query.\nHere are the tools available in JSONSchema format:' ~ '\n' }}
|
| 133 |
+
{{- '\n' ~ '<tools>' ~ '\n' }}
|
| 134 |
+
{{- render_tool_namespace("functions", tools) }}
|
| 135 |
+
{{- '</tools>' ~ '\n\n' }}
|
| 136 |
+
{{- 'To call tools, wrap all invocations in a single ' ~ toolcall_begin_token ~ toolcall_end_token ~ ' block. Parameter values containing nested objects or arrays are recursively expanded into XML elements. Example:\n' }}
|
| 137 |
+
{{- '\n' ~ toolcall_begin_token ~ '\n' }}
|
| 138 |
+
{{- ns_token + '<invoke name="tool-name-1">' }}
|
| 139 |
+
{{- ns_token + '<param-1>value-1' + ns_token + '</param-1>' }}
|
| 140 |
+
{{- ns_token + '<param-2>' }}
|
| 141 |
+
{{- ns_token + '<item>' }}
|
| 142 |
+
{{- ns_token + '<key-a>val-a' + ns_token + '</key-a>' }}
|
| 143 |
+
{{- ns_token + '<key-b>val-b' + ns_token + '</key-b>' }}
|
| 144 |
+
{{- ns_token + '</item>' }}
|
| 145 |
+
{{- ns_token + '</param-2>' }}
|
| 146 |
+
{{- ns_token + '</invoke>\n' }}
|
| 147 |
+
{{- ns_token + '<invoke name="tool-name-2">' }}
|
| 148 |
+
{{- ns_token + '<param-1>value-1' + ns_token + '</param-1>' }}
|
| 149 |
+
{{- ns_token + '</invoke>\n' }}
|
| 150 |
+
{{- toolcall_end_token }}
|
| 151 |
+
{%- endif -%}
|
| 152 |
+
{{- eos_token ~ '\n' }}
|
| 153 |
+
|
| 154 |
+
{#- Render messages -#}
|
| 155 |
+
{%- set last_tool_call = namespace(name=none) -%}
|
| 156 |
+
{%- for message in conversation_messages -%}
|
| 157 |
+
{%- if message.role == 'assistant' -%}
|
| 158 |
+
{{- bos_token ~ 'ai' ~ '\n' }}
|
| 159 |
+
|
| 160 |
+
{%- set reasoning_content = '' %}
|
| 161 |
+
{%- set content = visible_text(message.content) %}
|
| 162 |
+
{%- if message.reasoning_content is string %}
|
| 163 |
+
{%- set reasoning_content = message.reasoning_content %}
|
| 164 |
+
{%- else %}
|
| 165 |
+
{%- if think_end_token in content %}
|
| 166 |
+
{%- set reasoning_content = content.split(think_end_token)[0].strip('\n').split(think_begin_token)[-1].strip('\n') %}
|
| 167 |
+
{%- set content = content.split(think_end_token)[-1].strip('\n') %}
|
| 168 |
+
{%- endif %}
|
| 169 |
+
{%- endif %}
|
| 170 |
+
|
| 171 |
+
{%- if reasoning_content -%}
|
| 172 |
+
{#- Render thinking for every assistant turn (all-turn visible) -#}
|
| 173 |
+
{{- think_begin_token ~ reasoning_content ~ think_end_token }}
|
| 174 |
+
{%- else -%}
|
| 175 |
+
{#- No thinking rendered → prefix with think_end_token -#}
|
| 176 |
+
{{- think_end_token }}
|
| 177 |
+
{%- endif -%}
|
| 178 |
+
|
| 179 |
+
{%- if content -%}
|
| 180 |
+
{{- content }}
|
| 181 |
+
{%- endif -%}
|
| 182 |
+
{%- if message.tool_calls -%}
|
| 183 |
+
{{- toolcall_begin_token ~ '\n' }}
|
| 184 |
+
|
| 185 |
+
{%- for tool_call in message.tool_calls -%}
|
| 186 |
+
{%- if tool_call.function -%}
|
| 187 |
+
{%- set tool_call = tool_call.function -%}
|
| 188 |
+
{%- endif -%}
|
| 189 |
+
{{- ns_token + '<invoke name="' + tool_call.name + '">' }}
|
| 190 |
+
{%- set _args = tool_call.arguments -%}
|
| 191 |
+
{%- for k, v in _args.items() if v is not none %}
|
| 192 |
+
{{- ns_token + '<' + k + '>' -}}
|
| 193 |
+
{{- to_xml(v, ns_token) -}}
|
| 194 |
+
{{- ns_token + '</' + k + '>' }}
|
| 195 |
+
{%- endfor -%}
|
| 196 |
+
{{- ns_token + '</invoke>' ~ '\n' }}
|
| 197 |
+
{%- endfor -%}
|
| 198 |
+
|
| 199 |
+
{{- toolcall_end_token }}
|
| 200 |
+
{%- if message.tool_calls[-1].function -%}
|
| 201 |
+
{%- set last_tool_call.name = message.tool_calls[-1].function.name -%}
|
| 202 |
+
{%- else -%}
|
| 203 |
+
{%- set last_tool_call.name = message.tool_calls[-1].name -%}
|
| 204 |
+
{%- endif -%}
|
| 205 |
+
{%- else -%}
|
| 206 |
+
{%- set last_tool_call.name = none -%}
|
| 207 |
+
{%- endif -%}
|
| 208 |
+
{{- eos_token ~ '\n' }}
|
| 209 |
+
|
| 210 |
+
{%- elif message.role == 'tool' -%}
|
| 211 |
+
{%- if last_tool_call.name is none -%}
|
| 212 |
+
{{- raise_exception("Message has tool role, but there was no previous assistant message with a tool call!") }}
|
| 213 |
+
{%- endif -%}
|
| 214 |
+
{%- if loop.first or (conversation_messages[loop.index0 - 1].role != 'tool') -%}
|
| 215 |
+
{{- bos_token ~ 'tool' }}
|
| 216 |
+
{%- endif -%}
|
| 217 |
+
{{- '\n<response>' }}
|
| 218 |
+
{%- if message.content is string -%}
|
| 219 |
+
{{- message.content }}
|
| 220 |
+
{%- else -%}
|
| 221 |
+
{%- for tr in message.content -%}
|
| 222 |
+
{%- if tr is mapping and tr.type is defined and tr.type == 'image' -%}
|
| 223 |
+
{{- image_token }}
|
| 224 |
+
{%- elif tr is mapping and tr.type is defined and tr.type == 'video' -%}
|
| 225 |
+
{{- video_token }}
|
| 226 |
+
{%- else -%}
|
| 227 |
+
{{- tr.output if tr.output is defined else (tr.text if tr.type == 'text' and tr.text is defined else tr) }}
|
| 228 |
+
{%- endif -%}
|
| 229 |
+
{%- endfor -%}
|
| 230 |
+
{%- endif -%}
|
| 231 |
+
{{- '</response>' }}
|
| 232 |
+
{%- if loop.last or (conversation_messages[loop.index0 + 1].role != 'tool') -%}
|
| 233 |
+
{{- eos_token ~ '\n' -}}
|
| 234 |
+
{%- endif -%}
|
| 235 |
+
|
| 236 |
+
{%- elif message.role == 'user' -%}
|
| 237 |
+
{{- bos_token ~ 'user' ~ '\n' }}
|
| 238 |
+
{{- visible_text(message.content) }}
|
| 239 |
+
{{- eos_token ~ '\n' }}
|
| 240 |
+
{%- endif -%}
|
| 241 |
+
{%- endfor -%}
|
| 242 |
+
|
| 243 |
+
{#- Generation prompt -#}
|
| 244 |
+
{%- if add_generation_prompt -%}
|
| 245 |
+
{{- bos_token ~ 'ai' ~ '\n' }}
|
| 246 |
+
{%- if thinking_mode is defined and thinking_mode == "disabled" -%}
|
| 247 |
+
{{- think_end_token }}
|
| 248 |
+
{%- elif thinking_mode is defined and thinking_mode == "adaptive" -%}
|
| 249 |
+
{#- adaptive: no prefix, let model decide -#}
|
| 250 |
+
{%- elif thinking_mode is defined and thinking_mode == "enabled" -%}
|
| 251 |
+
{#- enabled or not defined: default to think -#}
|
| 252 |
+
{{- think_begin_token }}
|
| 253 |
+
{%- else -%}
|
| 254 |
+
{#- adaptive: no prefix, let model decide -#}
|
| 255 |
+
{%- endif -%}
|
| 256 |
+
{%- endif -%}
|
config.json
ADDED
|
@@ -0,0 +1,305 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"architectures": [
|
| 3 |
+
"MiniMaxM3SparseForConditionalGeneration"
|
| 4 |
+
],
|
| 5 |
+
"auto_map": {
|
| 6 |
+
"AutoConfig": "configuration_minimax_m3_vl.MiniMaxM3VLConfig"
|
| 7 |
+
},
|
| 8 |
+
"model_type": "minimax_m3_vl",
|
| 9 |
+
"text_config": {
|
| 10 |
+
"dtype": "bfloat16",
|
| 11 |
+
"hidden_size": 6144,
|
| 12 |
+
"intermediate_size": 3072,
|
| 13 |
+
"num_hidden_layers": 60,
|
| 14 |
+
"num_attention_heads": 64,
|
| 15 |
+
"num_key_value_heads": 4,
|
| 16 |
+
"head_dim": 128,
|
| 17 |
+
"vocab_size": 200064,
|
| 18 |
+
"max_position_embeddings": 524288,
|
| 19 |
+
"rms_norm_eps": 1e-06,
|
| 20 |
+
"use_gemma_norm": true,
|
| 21 |
+
"attention_output_gate": false,
|
| 22 |
+
"rope_theta": 5000000,
|
| 23 |
+
"rotary_dim": 64,
|
| 24 |
+
"partial_rotary_factor": 0.5,
|
| 25 |
+
"hidden_act": "swigluoai",
|
| 26 |
+
"use_qk_norm": true,
|
| 27 |
+
"tie_word_embeddings": false,
|
| 28 |
+
"dense_intermediate_size": 12288,
|
| 29 |
+
"shared_intermediate_size": 3072,
|
| 30 |
+
"num_local_experts": 128,
|
| 31 |
+
"num_experts_per_tok": 4,
|
| 32 |
+
"n_shared_experts": 1,
|
| 33 |
+
"scoring_func": "sigmoid",
|
| 34 |
+
"use_routing_bias": true,
|
| 35 |
+
"moe_layer_freq": [
|
| 36 |
+
0,
|
| 37 |
+
0,
|
| 38 |
+
0,
|
| 39 |
+
1,
|
| 40 |
+
1,
|
| 41 |
+
1,
|
| 42 |
+
1,
|
| 43 |
+
1,
|
| 44 |
+
1,
|
| 45 |
+
1,
|
| 46 |
+
1,
|
| 47 |
+
1,
|
| 48 |
+
1,
|
| 49 |
+
1,
|
| 50 |
+
1,
|
| 51 |
+
1,
|
| 52 |
+
1,
|
| 53 |
+
1,
|
| 54 |
+
1,
|
| 55 |
+
1,
|
| 56 |
+
1,
|
| 57 |
+
1,
|
| 58 |
+
1,
|
| 59 |
+
1,
|
| 60 |
+
1,
|
| 61 |
+
1,
|
| 62 |
+
1,
|
| 63 |
+
1,
|
| 64 |
+
1,
|
| 65 |
+
1,
|
| 66 |
+
1,
|
| 67 |
+
1,
|
| 68 |
+
1,
|
| 69 |
+
1,
|
| 70 |
+
1,
|
| 71 |
+
1,
|
| 72 |
+
1,
|
| 73 |
+
1,
|
| 74 |
+
1,
|
| 75 |
+
1,
|
| 76 |
+
1,
|
| 77 |
+
1,
|
| 78 |
+
1,
|
| 79 |
+
1,
|
| 80 |
+
1,
|
| 81 |
+
1,
|
| 82 |
+
1,
|
| 83 |
+
1,
|
| 84 |
+
1,
|
| 85 |
+
1,
|
| 86 |
+
1,
|
| 87 |
+
1,
|
| 88 |
+
1,
|
| 89 |
+
1,
|
| 90 |
+
1,
|
| 91 |
+
1,
|
| 92 |
+
1,
|
| 93 |
+
1,
|
| 94 |
+
1,
|
| 95 |
+
1
|
| 96 |
+
],
|
| 97 |
+
"qk_norm_type": "per_head",
|
| 98 |
+
"num_mtp_modules": 1,
|
| 99 |
+
"swiglu_alpha": 1.702,
|
| 100 |
+
"swiglu_limit": 7.0,
|
| 101 |
+
"routed_scaling_factor": 2.0,
|
| 102 |
+
"sparse_attention_config": {
|
| 103 |
+
"use_sparse_attention": true,
|
| 104 |
+
"sparse_index_dim": 128,
|
| 105 |
+
"sparse_num_index_heads": 4,
|
| 106 |
+
"sparse_topk_blocks": 16,
|
| 107 |
+
"sparse_block_size": 128,
|
| 108 |
+
"sparse_disable_index_value": [
|
| 109 |
+
0,
|
| 110 |
+
0,
|
| 111 |
+
0,
|
| 112 |
+
1,
|
| 113 |
+
1,
|
| 114 |
+
1,
|
| 115 |
+
1,
|
| 116 |
+
1,
|
| 117 |
+
1,
|
| 118 |
+
1,
|
| 119 |
+
1,
|
| 120 |
+
1,
|
| 121 |
+
1,
|
| 122 |
+
1,
|
| 123 |
+
1,
|
| 124 |
+
1,
|
| 125 |
+
1,
|
| 126 |
+
1,
|
| 127 |
+
1,
|
| 128 |
+
1,
|
| 129 |
+
1,
|
| 130 |
+
1,
|
| 131 |
+
1,
|
| 132 |
+
1,
|
| 133 |
+
1,
|
| 134 |
+
1,
|
| 135 |
+
1,
|
| 136 |
+
1,
|
| 137 |
+
1,
|
| 138 |
+
1,
|
| 139 |
+
1,
|
| 140 |
+
1,
|
| 141 |
+
1,
|
| 142 |
+
1,
|
| 143 |
+
1,
|
| 144 |
+
1,
|
| 145 |
+
1,
|
| 146 |
+
1,
|
| 147 |
+
1,
|
| 148 |
+
1,
|
| 149 |
+
1,
|
| 150 |
+
1,
|
| 151 |
+
1,
|
| 152 |
+
1,
|
| 153 |
+
1,
|
| 154 |
+
1,
|
| 155 |
+
1,
|
| 156 |
+
1,
|
| 157 |
+
1,
|
| 158 |
+
1,
|
| 159 |
+
1,
|
| 160 |
+
1,
|
| 161 |
+
1,
|
| 162 |
+
1,
|
| 163 |
+
1,
|
| 164 |
+
1,
|
| 165 |
+
1,
|
| 166 |
+
1,
|
| 167 |
+
1,
|
| 168 |
+
1
|
| 169 |
+
],
|
| 170 |
+
"sparse_score_type": "max",
|
| 171 |
+
"sparse_init_block": 0,
|
| 172 |
+
"sparse_local_block": 1,
|
| 173 |
+
"sparse_attention_freq": [
|
| 174 |
+
0,
|
| 175 |
+
0,
|
| 176 |
+
0,
|
| 177 |
+
1,
|
| 178 |
+
1,
|
| 179 |
+
1,
|
| 180 |
+
1,
|
| 181 |
+
1,
|
| 182 |
+
1,
|
| 183 |
+
1,
|
| 184 |
+
1,
|
| 185 |
+
1,
|
| 186 |
+
1,
|
| 187 |
+
1,
|
| 188 |
+
1,
|
| 189 |
+
1,
|
| 190 |
+
1,
|
| 191 |
+
1,
|
| 192 |
+
1,
|
| 193 |
+
1,
|
| 194 |
+
1,
|
| 195 |
+
1,
|
| 196 |
+
1,
|
| 197 |
+
1,
|
| 198 |
+
1,
|
| 199 |
+
1,
|
| 200 |
+
1,
|
| 201 |
+
1,
|
| 202 |
+
1,
|
| 203 |
+
1,
|
| 204 |
+
1,
|
| 205 |
+
1,
|
| 206 |
+
1,
|
| 207 |
+
1,
|
| 208 |
+
1,
|
| 209 |
+
1,
|
| 210 |
+
1,
|
| 211 |
+
1,
|
| 212 |
+
1,
|
| 213 |
+
1,
|
| 214 |
+
1,
|
| 215 |
+
1,
|
| 216 |
+
1,
|
| 217 |
+
1,
|
| 218 |
+
1,
|
| 219 |
+
1,
|
| 220 |
+
1,
|
| 221 |
+
1,
|
| 222 |
+
1,
|
| 223 |
+
1,
|
| 224 |
+
1,
|
| 225 |
+
1,
|
| 226 |
+
1,
|
| 227 |
+
1,
|
| 228 |
+
1,
|
| 229 |
+
1,
|
| 230 |
+
1,
|
| 231 |
+
1,
|
| 232 |
+
1,
|
| 233 |
+
1
|
| 234 |
+
]
|
| 235 |
+
},
|
| 236 |
+
"architectures": [
|
| 237 |
+
"MiniMaxM3SparseForCausalLM"
|
| 238 |
+
]
|
| 239 |
+
},
|
| 240 |
+
"vision_config": {
|
| 241 |
+
"hidden_size": 1280,
|
| 242 |
+
"num_attention_heads": 16,
|
| 243 |
+
"num_hidden_layers": 32,
|
| 244 |
+
"intermediate_size": 5120,
|
| 245 |
+
"patch_size": 14,
|
| 246 |
+
"image_size": 2016,
|
| 247 |
+
"projection_dim": 6144,
|
| 248 |
+
"position_embedding_type": "rope",
|
| 249 |
+
"rope_mode": "3d",
|
| 250 |
+
"rope_theta": 10000.0,
|
| 251 |
+
"attention_dropout": 0.0,
|
| 252 |
+
"hidden_act": "gelu",
|
| 253 |
+
"initializer_factor": 1.0,
|
| 254 |
+
"initializer_range": 0.02,
|
| 255 |
+
"layer_norm_eps": 1e-05,
|
| 256 |
+
"model_type": "clip_vision_model",
|
| 257 |
+
"num_channels": 3,
|
| 258 |
+
"vocab_size": 32000,
|
| 259 |
+
"img_token_compression_config": {
|
| 260 |
+
"image_token_compression_method": "patch_merge",
|
| 261 |
+
"spatial_merge_size": 2,
|
| 262 |
+
"temporal_patch_size": 2
|
| 263 |
+
},
|
| 264 |
+
"vision_segment_max_frames": 4
|
| 265 |
+
},
|
| 266 |
+
"img_token_compression_config": {
|
| 267 |
+
"image_token_compression_method": "patch_merge",
|
| 268 |
+
"spatial_merge_size": 2,
|
| 269 |
+
"temporal_patch_size": 2
|
| 270 |
+
},
|
| 271 |
+
"image_grid_pinpoints": "[(336, 336), (336, 672), (336, 1008), (336, 1344), (336, 1680), (336, 2016), (672, 336), (672, 672), (672, 1008), (672, 1344), (672, 1680), (672, 2016), (1008, 336), (1008, 672), (1008, 1008), (1008, 1344), (1008, 1680), (1008, 2016), (1344, 336), (1344, 672), (1344, 1008), (1344, 1344), (1344, 1680), (1344, 2016), (1680, 336), (1680, 672), (1680, 1008), (1680, 1344), (1680, 1680), (1680, 2016), (2016, 336), (2016, 672), (2016, 1008), (2016, 1344), (2016, 1680), (2016, 2016)]",
|
| 272 |
+
"image_seq_length": 576,
|
| 273 |
+
"image_token_index": 200025,
|
| 274 |
+
"video_token_index": 200026,
|
| 275 |
+
"multimodal_projector_bias": true,
|
| 276 |
+
"num_reward_heads": 0,
|
| 277 |
+
"process_image_mode": "dynamic_res",
|
| 278 |
+
"projector_hidden_act": "gelu",
|
| 279 |
+
"vision_feature_layer": -1,
|
| 280 |
+
"vision_feature_select_strategy": "full",
|
| 281 |
+
"torch_dtype": "bfloat16",
|
| 282 |
+
"transformers_version": "4.52.4",
|
| 283 |
+
"projector_hidden_size": 6144,
|
| 284 |
+
"quantization_config": {
|
| 285 |
+
"quant_algo": "NVFP4",
|
| 286 |
+
"group_size": 16,
|
| 287 |
+
"kv_cache_scheme": null,
|
| 288 |
+
"ignore": [
|
| 289 |
+
"lm_head",
|
| 290 |
+
"model.embed_tokens",
|
| 291 |
+
"vision_tower*",
|
| 292 |
+
"multi_modal_projector*",
|
| 293 |
+
"patch_merge_mlp*",
|
| 294 |
+
"*block_sparse_moe.gate"
|
| 295 |
+
],
|
| 296 |
+
"exclude_modules": [
|
| 297 |
+
"lm_head",
|
| 298 |
+
"model.embed_tokens",
|
| 299 |
+
"vision_tower*",
|
| 300 |
+
"multi_modal_projector*",
|
| 301 |
+
"patch_merge_mlp*",
|
| 302 |
+
"*block_sparse_moe.gate"
|
| 303 |
+
]
|
| 304 |
+
}
|
| 305 |
+
}
|
configuration_minimax_m3_vl.py
ADDED
|
@@ -0,0 +1,111 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
"""HuggingFace configs for the MiniMax VL family (M2 VL / M3 VL).
|
| 2 |
+
|
| 3 |
+
This file is bundled into every converted HF checkpoint so that loading via
|
| 4 |
+
``AutoConfig.from_pretrained(..., trust_remote_code=True)`` works without any
|
| 5 |
+
runtime dependency on sglang or other internal packages — only stock
|
| 6 |
+
``transformers`` is required.
|
| 7 |
+
|
| 8 |
+
The class definitions intentionally mirror
|
| 9 |
+
``sglang.srt.configs.minimax_vl``; if either side changes, keep them in sync.
|
| 10 |
+
|
| 11 |
+
The file is named ``configuration_minimax_m3_vl.py`` (matching the legacy
|
| 12 |
+
``model_type="minimax_m3_vl"`` and the converter's ``auto_map`` entry) so
|
| 13 |
+
that ckpts produced by this converter remain loadable by older sglang versions
|
| 14 |
+
that only know the ``MiniMaxM3VL*`` names. The canonical class is
|
| 15 |
+
``MiniMaxM3VLConfig``; ``MiniMaxM3VLConfig`` is a thin BC alias whose only
|
| 16 |
+
purpose is to be referenced from ``auto_map``.
|
| 17 |
+
"""
|
| 18 |
+
|
| 19 |
+
from typing import Optional
|
| 20 |
+
|
| 21 |
+
from transformers.configuration_utils import PretrainedConfig
|
| 22 |
+
from transformers.models.auto import CONFIG_MAPPING
|
| 23 |
+
|
| 24 |
+
|
| 25 |
+
def _coerce_sub_config(
|
| 26 |
+
sub_config: Optional[dict], default_model_type: str
|
| 27 |
+
) -> Optional[PretrainedConfig]:
|
| 28 |
+
"""Convert a config dict to a ``PretrainedConfig`` instance.
|
| 29 |
+
|
| 30 |
+
If ``model_type`` is registered in HF ``CONFIG_MAPPING`` the corresponding
|
| 31 |
+
config class is used; otherwise we fall back to a generic
|
| 32 |
+
``PretrainedConfig`` so all dict keys still become real attributes (M3's
|
| 33 |
+
text backbone uses ``model_type="minimax_m2"`` which is not in
|
| 34 |
+
``CONFIG_MAPPING``).
|
| 35 |
+
"""
|
| 36 |
+
if not isinstance(sub_config, dict):
|
| 37 |
+
return sub_config
|
| 38 |
+
model_type = sub_config.get("model_type", default_model_type)
|
| 39 |
+
cls = CONFIG_MAPPING.get(model_type, PretrainedConfig)
|
| 40 |
+
return cls(**sub_config)
|
| 41 |
+
|
| 42 |
+
|
| 43 |
+
class MiniMaxVLBaseConfig(PretrainedConfig):
|
| 44 |
+
"""Base config shared by every MiniMax VL variant.
|
| 45 |
+
|
| 46 |
+
Handles vision/text sub-config coercion. Concrete subclasses only need to
|
| 47 |
+
declare a unique ``model_type`` string.
|
| 48 |
+
"""
|
| 49 |
+
|
| 50 |
+
def __init__(
|
| 51 |
+
self,
|
| 52 |
+
vision_config: Optional[dict] = None,
|
| 53 |
+
text_config: Optional[dict] = None,
|
| 54 |
+
image_token_index: int = 200025,
|
| 55 |
+
video_token_index: int = 200026,
|
| 56 |
+
image_seq_length: int = 576,
|
| 57 |
+
process_image_mode: str = "dynamic_res",
|
| 58 |
+
projector_hidden_act: str = "gelu",
|
| 59 |
+
multimodal_projector_bias: bool = True,
|
| 60 |
+
vision_feature_layer: int = -1,
|
| 61 |
+
vision_feature_select_strategy: str = "full",
|
| 62 |
+
img_token_compression_config: Optional[dict] = None,
|
| 63 |
+
image_grid_pinpoints: Optional[str] = None,
|
| 64 |
+
**kwargs,
|
| 65 |
+
):
|
| 66 |
+
self.vision_config = _coerce_sub_config(vision_config, "clip_vision_model")
|
| 67 |
+
self.text_config = _coerce_sub_config(text_config, "mixtral")
|
| 68 |
+
|
| 69 |
+
self.image_token_index = image_token_index
|
| 70 |
+
self.video_token_index = video_token_index
|
| 71 |
+
self.image_seq_length = image_seq_length
|
| 72 |
+
self.process_image_mode = process_image_mode
|
| 73 |
+
self.projector_hidden_act = projector_hidden_act
|
| 74 |
+
self.multimodal_projector_bias = multimodal_projector_bias
|
| 75 |
+
self.vision_feature_layer = vision_feature_layer
|
| 76 |
+
self.vision_feature_select_strategy = vision_feature_select_strategy
|
| 77 |
+
self.img_token_compression_config = img_token_compression_config or {}
|
| 78 |
+
self.image_grid_pinpoints = image_grid_pinpoints
|
| 79 |
+
|
| 80 |
+
super().__init__(**kwargs)
|
| 81 |
+
|
| 82 |
+
def __post_init__(self, **kwargs):
|
| 83 |
+
super().__post_init__(**kwargs)
|
| 84 |
+
if hasattr(self, "vision_config"):
|
| 85 |
+
self.vision_config = _coerce_sub_config(self.vision_config, "clip_vision_model")
|
| 86 |
+
if hasattr(self, "text_config"):
|
| 87 |
+
self.text_config = _coerce_sub_config(self.text_config, "mixtral")
|
| 88 |
+
|
| 89 |
+
|
| 90 |
+
class MiniMaxM2VLConfig(MiniMaxVLBaseConfig):
|
| 91 |
+
"""MiniMax M2 VL: vision tower + M2 (Mixtral-style MoE) text backbone."""
|
| 92 |
+
|
| 93 |
+
model_type = "minimax_m2_vl"
|
| 94 |
+
|
| 95 |
+
|
| 96 |
+
class MiniMaxM3VLConfig(MiniMaxVLBaseConfig):
|
| 97 |
+
"""MiniMax M3 VL: vision tower + M3 (mixed sparse/dense MoE) text backbone."""
|
| 98 |
+
|
| 99 |
+
model_type = "minimax_m3_vl"
|
| 100 |
+
|
| 101 |
+
|
| 102 |
+
class MiniMaxM2MiniVLConfig(MiniMaxM2VLConfig):
|
| 103 |
+
"""Legacy alias kept so old ``model_type="minimax_m2_mini_vl"`` ckpts load."""
|
| 104 |
+
|
| 105 |
+
model_type = "minimax_m2_mini_vl"
|
| 106 |
+
|
| 107 |
+
|
| 108 |
+
class MiniMaxM3VLConfig(MiniMaxM3VLConfig):
|
| 109 |
+
"""Legacy alias kept so old ``model_type="minimax_m3_vl"`` ckpts load."""
|
| 110 |
+
|
| 111 |
+
model_type = "minimax_m3_vl"
|
generation_config.json
ADDED
|
@@ -0,0 +1,13 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"chat_template_file": "chat_template.jinja",
|
| 3 |
+
"chat_template": "chat_template.jinja",
|
| 4 |
+
"reasoning_parser": "minimax-m3",
|
| 5 |
+
"tool_call_parser": "minimax-m3",
|
| 6 |
+
"model_id": "sparkarena/Minimax-M3-v0-NVFP4",
|
| 7 |
+
"base_model_id": "MiniMaxAI/Minimax-M3",
|
| 8 |
+
"bos_token_id": 200019,
|
| 9 |
+
"do_sample": true,
|
| 10 |
+
"eos_token_id": 200020,
|
| 11 |
+
"temperature": 1.0,
|
| 12 |
+
"top_p": 0.95
|
| 13 |
+
}
|
image_processor.py
ADDED
|
@@ -0,0 +1,223 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Copyright 2023-2024 SGLang Team
|
| 2 |
+
# Licensed under the Apache License, Version 2.0 (the "License");
|
| 3 |
+
"""
|
| 4 |
+
MiniMax VL family HuggingFace-compatible Processor, ImageProcessor, VideoProcessor.
|
| 5 |
+
"""
|
| 6 |
+
import math
|
| 7 |
+
from typing import List, Tuple
|
| 8 |
+
|
| 9 |
+
import torch
|
| 10 |
+
from torchvision.transforms import InterpolationMode
|
| 11 |
+
from transformers import BatchFeature
|
| 12 |
+
from transformers.image_processing_utils_fast import (
|
| 13 |
+
BaseImageProcessorFast,
|
| 14 |
+
group_images_by_shape,
|
| 15 |
+
reorder_images,
|
| 16 |
+
)
|
| 17 |
+
from transformers.image_utils import PILImageResampling, SizeDict
|
| 18 |
+
from transformers.processing_utils import (
|
| 19 |
+
ImagesKwargs,
|
| 20 |
+
Unpack,
|
| 21 |
+
)
|
| 22 |
+
from transformers.utils import TensorType
|
| 23 |
+
|
| 24 |
+
MAX_RATIO = 200
|
| 25 |
+
|
| 26 |
+
|
| 27 |
+
def round_by_factor(number: int, factor: int) -> int:
|
| 28 |
+
return round(number / factor) * factor
|
| 29 |
+
|
| 30 |
+
|
| 31 |
+
def ceil_by_factor(number: int, factor: int) -> int:
|
| 32 |
+
return math.ceil(number / factor) * factor
|
| 33 |
+
|
| 34 |
+
|
| 35 |
+
def floor_by_factor(number: int, factor: int) -> int:
|
| 36 |
+
return math.floor(number / factor) * factor
|
| 37 |
+
|
| 38 |
+
|
| 39 |
+
def smart_resize(
|
| 40 |
+
height: int,
|
| 41 |
+
width: int,
|
| 42 |
+
factor: int = 28,
|
| 43 |
+
min_pixels: int = 4 * 28 * 28,
|
| 44 |
+
max_pixels: int = 451584,
|
| 45 |
+
) -> tuple[int, int]:
|
| 46 |
+
if max(height, width) / min(height, width) > MAX_RATIO:
|
| 47 |
+
raise ValueError(
|
| 48 |
+
f"absolute aspect ratio must be smaller than {MAX_RATIO}, "
|
| 49 |
+
f"got {max(height, width) / min(height, width)}"
|
| 50 |
+
)
|
| 51 |
+
h_bar = max(factor, round_by_factor(height, factor))
|
| 52 |
+
w_bar = max(factor, round_by_factor(width, factor))
|
| 53 |
+
if h_bar * w_bar > max_pixels:
|
| 54 |
+
beta = math.sqrt((height * width) / max_pixels)
|
| 55 |
+
h_bar = floor_by_factor(height / beta, factor)
|
| 56 |
+
w_bar = floor_by_factor(width / beta, factor)
|
| 57 |
+
elif h_bar * w_bar < min_pixels:
|
| 58 |
+
beta = math.sqrt(min_pixels / (height * width))
|
| 59 |
+
h_bar = ceil_by_factor(height * beta, factor)
|
| 60 |
+
w_bar = ceil_by_factor(width * beta, factor)
|
| 61 |
+
return h_bar, w_bar
|
| 62 |
+
|
| 63 |
+
|
| 64 |
+
# ==============================================================================
|
| 65 |
+
# MiniMax M3 VL Image Processor Fast (Fast Mode - Torch based)
|
| 66 |
+
# ==============================================================================
|
| 67 |
+
|
| 68 |
+
|
| 69 |
+
class MiniMaxM3VLImageProcessorKwargs(ImagesKwargs, total=False):
|
| 70 |
+
patch_size: int
|
| 71 |
+
temporal_patch_size: int
|
| 72 |
+
merge_size: int
|
| 73 |
+
max_pixels: int
|
| 74 |
+
|
| 75 |
+
|
| 76 |
+
class MiniMaxM3VLImageProcessor(BaseImageProcessorFast):
|
| 77 |
+
do_resize = True
|
| 78 |
+
resample = PILImageResampling.BICUBIC
|
| 79 |
+
size = {"height": 672, "width": 672} # required by base class validation, not used as resize bound
|
| 80 |
+
default_to_square = False
|
| 81 |
+
do_rescale = True
|
| 82 |
+
rescale_factor = 1 / 255
|
| 83 |
+
do_normalize = True
|
| 84 |
+
image_mean = [0.48145466, 0.4578275, 0.40821073]
|
| 85 |
+
image_std = [0.26862954, 0.26130258, 0.27577711]
|
| 86 |
+
do_convert_rgb = True
|
| 87 |
+
patch_size = 14
|
| 88 |
+
temporal_patch_size = 2
|
| 89 |
+
merge_size = 2
|
| 90 |
+
max_pixels = 451584 # 672*672
|
| 91 |
+
valid_kwargs = MiniMaxM3VLImageProcessorKwargs
|
| 92 |
+
model_input_names = ["pixel_values", "image_grid_thw"]
|
| 93 |
+
|
| 94 |
+
def __init__(self, **kwargs: Unpack[MiniMaxM3VLImageProcessorKwargs]):
|
| 95 |
+
super().__init__(**kwargs)
|
| 96 |
+
|
| 97 |
+
def preprocess(
|
| 98 |
+
self, images, **kwargs: Unpack[MiniMaxM3VLImageProcessorKwargs]
|
| 99 |
+
) -> BatchFeature:
|
| 100 |
+
return super().preprocess(images, **kwargs)
|
| 101 |
+
|
| 102 |
+
def _preprocess(
|
| 103 |
+
self,
|
| 104 |
+
images: List[torch.Tensor],
|
| 105 |
+
do_resize: bool,
|
| 106 |
+
size: SizeDict,
|
| 107 |
+
resample: PILImageResampling | InterpolationMode | int | None,
|
| 108 |
+
do_rescale: bool,
|
| 109 |
+
rescale_factor: float,
|
| 110 |
+
do_normalize: bool,
|
| 111 |
+
image_mean: float | List[float] | None,
|
| 112 |
+
image_std: float | List[float] | None,
|
| 113 |
+
patch_size: int,
|
| 114 |
+
temporal_patch_size: int,
|
| 115 |
+
merge_size: int,
|
| 116 |
+
max_pixels: int,
|
| 117 |
+
disable_grouping: bool | None,
|
| 118 |
+
return_tensors: str | TensorType | None,
|
| 119 |
+
**kwargs,
|
| 120 |
+
) -> BatchFeature:
|
| 121 |
+
grouped_images, grouped_images_index = group_images_by_shape(
|
| 122 |
+
images, disable_grouping=disable_grouping
|
| 123 |
+
)
|
| 124 |
+
resized_images_grouped = {}
|
| 125 |
+
factor = patch_size * merge_size
|
| 126 |
+
for shape, stacked_images in grouped_images.items():
|
| 127 |
+
height, width = stacked_images.shape[-2:]
|
| 128 |
+
if do_resize:
|
| 129 |
+
resized_height, resized_width = smart_resize(
|
| 130 |
+
height, width, factor=factor,
|
| 131 |
+
max_pixels=max_pixels,
|
| 132 |
+
)
|
| 133 |
+
stacked_images = self.resize(
|
| 134 |
+
stacked_images,
|
| 135 |
+
size=SizeDict(height=resized_height, width=resized_width),
|
| 136 |
+
resample=resample,
|
| 137 |
+
)
|
| 138 |
+
resized_images_grouped[shape] = stacked_images
|
| 139 |
+
|
| 140 |
+
resized_images = reorder_images(resized_images_grouped, grouped_images_index)
|
| 141 |
+
|
| 142 |
+
grouped_images, grouped_images_index = group_images_by_shape(
|
| 143 |
+
resized_images, disable_grouping=disable_grouping
|
| 144 |
+
)
|
| 145 |
+
processed_images_grouped = {}
|
| 146 |
+
processed_grids = {}
|
| 147 |
+
|
| 148 |
+
for shape, stacked_images in grouped_images.items():
|
| 149 |
+
resized_height, resized_width = stacked_images.shape[-2:]
|
| 150 |
+
|
| 151 |
+
patches = self.rescale_and_normalize(
|
| 152 |
+
stacked_images,
|
| 153 |
+
do_rescale,
|
| 154 |
+
rescale_factor,
|
| 155 |
+
do_normalize,
|
| 156 |
+
image_mean,
|
| 157 |
+
image_std,
|
| 158 |
+
)
|
| 159 |
+
if patches.ndim == 4:
|
| 160 |
+
patches = patches.unsqueeze(1)
|
| 161 |
+
|
| 162 |
+
if patches.shape[1] % temporal_patch_size != 0:
|
| 163 |
+
repeats = patches[:, -1:].repeat(
|
| 164 |
+
1,
|
| 165 |
+
temporal_patch_size - (patches.shape[1] % temporal_patch_size),
|
| 166 |
+
1,
|
| 167 |
+
1,
|
| 168 |
+
1,
|
| 169 |
+
)
|
| 170 |
+
patches = torch.cat([patches, repeats], dim=1)
|
| 171 |
+
|
| 172 |
+
batch_size, grid_t, channel = patches.shape[:3]
|
| 173 |
+
grid_t = grid_t // temporal_patch_size
|
| 174 |
+
grid_h, grid_w = resized_height // patch_size, resized_width // patch_size
|
| 175 |
+
|
| 176 |
+
patches = patches.view(
|
| 177 |
+
batch_size,
|
| 178 |
+
grid_t,
|
| 179 |
+
temporal_patch_size,
|
| 180 |
+
channel,
|
| 181 |
+
grid_h // merge_size,
|
| 182 |
+
merge_size,
|
| 183 |
+
patch_size,
|
| 184 |
+
grid_w // merge_size,
|
| 185 |
+
merge_size,
|
| 186 |
+
patch_size,
|
| 187 |
+
)
|
| 188 |
+
patches = patches.permute(0, 1, 4, 7, 5, 8, 3, 2, 6, 9)
|
| 189 |
+
|
| 190 |
+
flatten_patches = patches.reshape(
|
| 191 |
+
batch_size,
|
| 192 |
+
grid_t * grid_h * grid_w,
|
| 193 |
+
channel * temporal_patch_size * patch_size * patch_size,
|
| 194 |
+
)
|
| 195 |
+
|
| 196 |
+
processed_images_grouped[shape] = flatten_patches
|
| 197 |
+
processed_grids[shape] = [[grid_t, grid_h, grid_w]] * batch_size
|
| 198 |
+
|
| 199 |
+
processed_images = reorder_images(
|
| 200 |
+
processed_images_grouped, grouped_images_index
|
| 201 |
+
)
|
| 202 |
+
processed_grids = reorder_images(processed_grids, grouped_images_index)
|
| 203 |
+
|
| 204 |
+
pixel_values = torch.cat(processed_images, dim=0)
|
| 205 |
+
image_grid_thw = torch.tensor(processed_grids, dtype=torch.long)
|
| 206 |
+
|
| 207 |
+
return BatchFeature(
|
| 208 |
+
data={"pixel_values": pixel_values, "image_grid_thw": image_grid_thw},
|
| 209 |
+
tensor_type=return_tensors,
|
| 210 |
+
)
|
| 211 |
+
|
| 212 |
+
def get_number_of_image_patches(self, height: int, width: int, images_kwargs=None):
|
| 213 |
+
images_kwargs = images_kwargs or {}
|
| 214 |
+
patch_size = images_kwargs.get("patch_size", self.patch_size)
|
| 215 |
+
merge_size = images_kwargs.get("merge_size", self.merge_size)
|
| 216 |
+
max_pixels = images_kwargs.get("max_pixels", self.max_pixels)
|
| 217 |
+
|
| 218 |
+
resized_height, resized_width = smart_resize(
|
| 219 |
+
height, width, factor=patch_size * merge_size,
|
| 220 |
+
max_pixels=max_pixels,
|
| 221 |
+
)
|
| 222 |
+
grid_h, grid_w = resized_height // patch_size, resized_width // patch_size
|
| 223 |
+
return grid_h * grid_w
|
merges.txt
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
model-00001-of-00046.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:122ab05ba5c86d1985578fb44611b95de51822c0fd4247d49394131b2df45fa6
|
| 3 |
+
size 5334465808
|
model-00003-of-00046.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c10100bdf6827e5712d8d276e0d1de04a417bab8a5a872f80456196ce878b1cb
|
| 3 |
+
size 5365482560
|
model-00004-of-00046.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:598b24b1cabc909a6e58c252eeaa8086d24809e9ef243229228d767cab6fd48a
|
| 3 |
+
size 5360457976
|
model-00005-of-00046.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:2b17173803b4e14db6ce3333fdbdb93f1648489c9fc858097685e1c40beecf15
|
| 3 |
+
size 5366807824
|
model-00006-of-00046.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:57eb97f4d304ad7678e88e96443c6980b3fe0f289ff2420250acaff93fdf5b32
|
| 3 |
+
size 5360458896
|
model-00007-of-00046.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:b5cf607dbb98e9878acb3fb40e2cc66459570971b0b71e1866d16cf96a5637b0
|
| 3 |
+
size 5365480144
|
model-00008-of-00046.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:2f2ce6ba229c5af4956ce5a539c464e48a0cec5cf82ead8de032664fe63e93a5
|
| 3 |
+
size 5366809424
|
model-00010-of-00046.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:ef13e9ccd8d66d2b21dd7e7c338fc1ce266eebdb3f78e8abacdd8a3b4efbec44
|
| 3 |
+
size 5366807088
|
model-00011-of-00046.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:8b5a7154166b66fb14dd79cfbf899f622e49c9bc4876a1b38882435ddf1e93db
|
| 3 |
+
size 5368569768
|
model-00012-of-00046.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4c30d8d7f8fb0a03ff5a04c5379e5764cf4b0ed85401cc4f906d180f052929e8
|
| 3 |
+
size 5367989008
|
model-00016-of-00046.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a8cbb367ef01a0fb43108f231519f6deeb900f03816d937426caff24af08666e
|
| 3 |
+
size 5360459992
|
model-00017-of-00046.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:ebb606ba2c011266155f92e39f9cc73349a5e88de50a4b11af36e949e0d8fd42
|
| 3 |
+
size 5366807848
|
model-00019-of-00046.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7731a1193d7edb3ccdb7ea7de50c56fc88d2d14374bafe7c97e284ed6b0b3819
|
| 3 |
+
size 5368570032
|
model-00021-of-00046.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:da8c3a7e530453d8b9d48d0d872b234e74bde9174280ed26b1f7964fd5bd19cf
|
| 3 |
+
size 5366811400
|
model-00022-of-00046.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:45ae32de2d734119051c4054e76b0108b1eede16246924e97239ff608dea86b0
|
| 3 |
+
size 5360459624
|
model-00023-of-00046.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:67da4379d407073f449bee7e0472d5227bec5775827e0e30bd1a9d66a811633d
|
| 3 |
+
size 5366808704
|
model-00024-of-00046.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c984c9c4274a8903213789d0857543086cc9aa1aace68b983ffac0216e51d8d9
|
| 3 |
+
size 5368569904
|
model-00025-of-00046.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e263cf654c44aa3ff8f8652ec6b6184e5a0ba7a93b99790ef18ca6fa23601c54
|
| 3 |
+
size 5367989928
|
model-00027-of-00046.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:dfa06b1ff1cae9e2829b535d5b56b7699d25ac3081b8d5cd56191ffcc8042192
|
| 3 |
+
size 5360459616
|
model-00029-of-00046.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c360c18e27aea8a95923d9e6301fa892fe9241f758882304be032cfd86d58887
|
| 3 |
+
size 5360459864
|
model-00030-of-00046.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:10315df007509b66b7a855b391ba1d8594dc951505a471d54a43093a481ae9e3
|
| 3 |
+
size 5366808768
|
model-00032-of-00046.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:9eeb434b14cddee164c9c9c0f0fba559974130f1216b822701f4b45780d8086d
|
| 3 |
+
size 5360459928
|
model-00033-of-00046.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a321c5690159476b5d8a559d59f77a5501a47e1f95c4faff6f68a9f02788d172
|
| 3 |
+
size 5366808712
|
model-00034-of-00046.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7d1c969fb68a8c2039c5961722013f5ae4ec789177feea9f6d3b8b3165d8097d
|
| 3 |
+
size 5360459680
|
model-00035-of-00046.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:47abd238d89a9a1f014fc1c51dc3e4eac91f39f216aa8ef8280e29efd6747a3c
|
| 3 |
+
size 5366809264
|
model-00038-of-00046.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:caf1c4caf35858149314eda40476c02d58fcd9a971310311df29aa8c401b7f35
|
| 3 |
+
size 5367991064
|
model-00041-of-00046.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:52b51b76b31c62f1e17de2f510e733afb8a1288181b39dd2a8faa10d40373f29
|
| 3 |
+
size 5365482736
|
model-00042-of-00046.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:9bd6f0995b8a9eac06057624d50087ddd19826d7ba4cca5d5fbb03739fc93971
|
| 3 |
+
size 5360459736
|
model-00043-of-00046.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e5e7351401bd3df070a1e9f39320ac62709327a06a3c13ea33a16f0f2663a96d
|
| 3 |
+
size 5366809736
|
model-00044-of-00046.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e985eb03cb250e3b09cf12b7b11dc963d00e1bb5d771471add75b57c6066b740
|
| 3 |
+
size 5366956312
|
model-00045-of-00046.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d9089f3efa184fe1bfaf511e29e92751ff4a9c41cfa72f1a2db8d0ce3e2f7a60
|
| 3 |
+
size 5365737864
|
model.safetensors.index.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
preprocessor_config.json
ADDED
|
@@ -0,0 +1,32 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"processor_class": "MiniMaxVLProcessor",
|
| 3 |
+
"auto_map": {
|
| 4 |
+
"AutoImageProcessor": "image_processor.MiniMaxM3VLImageProcessor",
|
| 5 |
+
"AutoProcessor": "processing_minimax.MiniMaxVLProcessor",
|
| 6 |
+
"AutoVideoProcessor": "video_processor.MiniMaxM3VLVideoProcessor"
|
| 7 |
+
},
|
| 8 |
+
"process_image_mode": "dynamic_res",
|
| 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 |
+
"size": [
|
| 20 |
+
672,
|
| 21 |
+
672
|
| 22 |
+
],
|
| 23 |
+
"patch_size": 14,
|
| 24 |
+
"img_token_compression_config": {
|
| 25 |
+
"image_token_compression_threshold": 1.1,
|
| 26 |
+
"image_token_compression_method": "patch_merge",
|
| 27 |
+
"max_image_resolution": 1008,
|
| 28 |
+
"spatial_merge_size": 2,
|
| 29 |
+
"temporal_patch_size": 2
|
| 30 |
+
},
|
| 31 |
+
"add_start_end_special_tokens": true
|
| 32 |
+
}
|
processing_minimax.py
ADDED
|
@@ -0,0 +1,254 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Copyright 2023-2024 SGLang Team
|
| 2 |
+
# Licensed under the Apache License, Version 2.0 (the "License");
|
| 3 |
+
"""
|
| 4 |
+
MiniMax VL family HuggingFace-compatible Processor, ImageProcessor, VideoProcessor.
|
| 5 |
+
"""
|
| 6 |
+
|
| 7 |
+
import math
|
| 8 |
+
import re
|
| 9 |
+
from typing import List, Optional, Tuple, Union
|
| 10 |
+
|
| 11 |
+
import torch
|
| 12 |
+
import torchvision
|
| 13 |
+
from torchvision.transforms import InterpolationMode
|
| 14 |
+
from transformers import BatchFeature
|
| 15 |
+
from transformers.image_processing_utils_fast import (
|
| 16 |
+
BaseImageProcessorFast,
|
| 17 |
+
group_images_by_shape,
|
| 18 |
+
reorder_images,
|
| 19 |
+
)
|
| 20 |
+
from transformers.image_utils import PILImageResampling, SizeDict
|
| 21 |
+
from transformers.processing_utils import (
|
| 22 |
+
ImagesKwargs,
|
| 23 |
+
ProcessingKwargs,
|
| 24 |
+
ProcessorMixin,
|
| 25 |
+
Unpack,
|
| 26 |
+
VideosKwargs,
|
| 27 |
+
)
|
| 28 |
+
from transformers.utils import TensorType
|
| 29 |
+
from transformers.video_processing_utils import BaseVideoProcessor
|
| 30 |
+
from transformers.video_utils import group_videos_by_shape, reorder_videos
|
| 31 |
+
|
| 32 |
+
|
| 33 |
+
class MiniMaxVLProcessorKwargs(ProcessingKwargs, total=False):
|
| 34 |
+
_defaults = {
|
| 35 |
+
"videos_kwargs": {
|
| 36 |
+
"do_resize": False,
|
| 37 |
+
"return_metadata": True,
|
| 38 |
+
},
|
| 39 |
+
}
|
| 40 |
+
|
| 41 |
+
|
| 42 |
+
class MiniMaxVLProcessor(ProcessorMixin):
|
| 43 |
+
IMAGE_TOKEN = "]<]image[>["
|
| 44 |
+
VIDEO_TOKEN = "]<]video[>["
|
| 45 |
+
VISION_START_TOKEN = "]<]start of image[>["
|
| 46 |
+
VISION_END_TOKEN = "]<]end of image[>["
|
| 47 |
+
|
| 48 |
+
def __init__(
|
| 49 |
+
self, image_processor=None, tokenizer=None, video_processor=None, **kwargs
|
| 50 |
+
):
|
| 51 |
+
self.image_token_id = tokenizer.convert_tokens_to_ids(self.IMAGE_TOKEN)
|
| 52 |
+
self.video_token_id = tokenizer.convert_tokens_to_ids(self.VIDEO_TOKEN)
|
| 53 |
+
super().__init__(image_processor, tokenizer, video_processor)
|
| 54 |
+
# Video expansion also uses image start/end tokens. Separate video
|
| 55 |
+
# start/end tokens exist in the tokenizer, but the original MiniMax
|
| 56 |
+
# serving path did not use them; keep that behavior for compatibility.
|
| 57 |
+
self.vision_start_token_id = tokenizer.convert_tokens_to_ids(
|
| 58 |
+
self.VISION_START_TOKEN
|
| 59 |
+
)
|
| 60 |
+
self.vision_end_token_id = tokenizer.convert_tokens_to_ids(
|
| 61 |
+
self.VISION_END_TOKEN
|
| 62 |
+
)
|
| 63 |
+
|
| 64 |
+
def _prune_video_tokens(
|
| 65 |
+
self,
|
| 66 |
+
input_text: str,
|
| 67 |
+
video_segments: List[int],
|
| 68 |
+
video_token: str,
|
| 69 |
+
) -> str:
|
| 70 |
+
"""
|
| 71 |
+
Prune video tokens by temporal_patch_size (e.g., 2:1).
|
| 72 |
+
|
| 73 |
+
Expects the prompt to carry exactly sum(video_segments) video
|
| 74 |
+
tokens — i.e. one token per *sampled* frame. Then drops token.
|
| 75 |
+
|
| 76 |
+
Args:
|
| 77 |
+
input_text: prompt with N video_tokens per segment
|
| 78 |
+
video_segments: actual sampled frame count per video segment
|
| 79 |
+
video_token: the video token string, e.g. ']<]video[>['
|
| 80 |
+
|
| 81 |
+
Returns:
|
| 82 |
+
Pruned input_text with ~N/temporal_patch_size tokens per segment.
|
| 83 |
+
"""
|
| 84 |
+
# If no videos or temporal_patch_size <= 1, no pruning needed
|
| 85 |
+
if not video_segments or self.video_processor.temporal_patch_size <= 1:
|
| 86 |
+
return input_text
|
| 87 |
+
|
| 88 |
+
# Split while keeping delimiters
|
| 89 |
+
special_tokens = [video_token] # , image_token]
|
| 90 |
+
pattern = "|".join(map(re.escape, special_tokens))
|
| 91 |
+
parts = re.split(f"({pattern})", input_text)
|
| 92 |
+
|
| 93 |
+
def is_timestamp(text: str) -> bool:
|
| 94 |
+
"""Check if text ends with timestamp format like ']<]0.0 seconds[>['"""
|
| 95 |
+
return (
|
| 96 |
+
text.endswith("seconds[>[")
|
| 97 |
+
or text.endswith("seconds[>[ ")
|
| 98 |
+
or text.endswith("seconds [>[")
|
| 99 |
+
or text.endswith("seconds [>[ ")
|
| 100 |
+
)
|
| 101 |
+
|
| 102 |
+
def extract_timestamp(text: str) -> str:
|
| 103 |
+
"""Extract timestamp text from the end, starting from ']<]'"""
|
| 104 |
+
start_index = text.rfind("]<]")
|
| 105 |
+
if start_index == -1:
|
| 106 |
+
raise ValueError(f"Failed to extract timestamp: {text}")
|
| 107 |
+
return text[start_index:]
|
| 108 |
+
|
| 109 |
+
# Build new text with pruned video tokens
|
| 110 |
+
final_parts = []
|
| 111 |
+
current_seg_idx = 0 # Which video segment we're in
|
| 112 |
+
frame_in_seg = 0 # Frame index within current segment
|
| 113 |
+
last_timestamp_len = 0 # Length of timestamp to potentially remove
|
| 114 |
+
|
| 115 |
+
for part in parts:
|
| 116 |
+
if part == video_token:
|
| 117 |
+
if current_seg_idx < len(video_segments):
|
| 118 |
+
if frame_in_seg % self.video_processor.temporal_patch_size == 0:
|
| 119 |
+
# Keep this video token
|
| 120 |
+
final_parts.append(part)
|
| 121 |
+
frame_in_seg += 1
|
| 122 |
+
if frame_in_seg >= video_segments[current_seg_idx]:
|
| 123 |
+
current_seg_idx += 1
|
| 124 |
+
frame_in_seg = 0
|
| 125 |
+
last_timestamp_len = 0
|
| 126 |
+
else:
|
| 127 |
+
# Skip this video token
|
| 128 |
+
frame_in_seg += 1
|
| 129 |
+
if frame_in_seg >= video_segments[current_seg_idx]:
|
| 130 |
+
current_seg_idx += 1
|
| 131 |
+
frame_in_seg = 0
|
| 132 |
+
# Remove the timestamp that was already appended
|
| 133 |
+
if last_timestamp_len > 0:
|
| 134 |
+
# Truncate the last part to remove timestamp
|
| 135 |
+
assert len(final_parts) > 0
|
| 136 |
+
final_parts[-1] = final_parts[-1][:-last_timestamp_len]
|
| 137 |
+
last_timestamp_len = 0
|
| 138 |
+
else:
|
| 139 |
+
# No more video segments, keep as is
|
| 140 |
+
final_parts.append(part)
|
| 141 |
+
last_timestamp_len = 0
|
| 142 |
+
else:
|
| 143 |
+
# Text part
|
| 144 |
+
final_parts.append(part)
|
| 145 |
+
# Check if this text ends with a timestamp
|
| 146 |
+
if is_timestamp(part):
|
| 147 |
+
last_timestamp_len = len(extract_timestamp(part))
|
| 148 |
+
else:
|
| 149 |
+
last_timestamp_len = 0
|
| 150 |
+
|
| 151 |
+
return "".join(final_parts)
|
| 152 |
+
|
| 153 |
+
def __call__(
|
| 154 |
+
self,
|
| 155 |
+
images=None,
|
| 156 |
+
text=None,
|
| 157 |
+
videos=None,
|
| 158 |
+
**kwargs: Unpack[MiniMaxVLProcessorKwargs],
|
| 159 |
+
) -> BatchFeature:
|
| 160 |
+
output_kwargs = self._merge_kwargs(
|
| 161 |
+
MiniMaxVLProcessorKwargs,
|
| 162 |
+
tokenizer_init_kwargs=self.tokenizer.init_kwargs,
|
| 163 |
+
**kwargs,
|
| 164 |
+
)
|
| 165 |
+
|
| 166 |
+
if images is not None:
|
| 167 |
+
images_kwargs = output_kwargs["images_kwargs"]
|
| 168 |
+
image_inputs = self.image_processor(images=images, **images_kwargs)
|
| 169 |
+
image_grid_thw = image_inputs["image_grid_thw"]
|
| 170 |
+
|
| 171 |
+
else:
|
| 172 |
+
image_inputs = {}
|
| 173 |
+
image_grid_thw = None
|
| 174 |
+
|
| 175 |
+
if videos is not None:
|
| 176 |
+
videos_kwargs = output_kwargs["videos_kwargs"]
|
| 177 |
+
video_inputs = self.video_processor(videos=videos, **videos_kwargs)
|
| 178 |
+
video_grid_thw = video_inputs["video_grid_thw"]
|
| 179 |
+
if not kwargs.get("return_metadata"):
|
| 180 |
+
video_metadata = video_inputs.pop("video_metadata")
|
| 181 |
+
else:
|
| 182 |
+
video_metadata = video_inputs["video_metadata"]
|
| 183 |
+
else:
|
| 184 |
+
video_inputs = {}
|
| 185 |
+
video_grid_thw = None
|
| 186 |
+
|
| 187 |
+
if not isinstance(text, list):
|
| 188 |
+
text = [text]
|
| 189 |
+
text = text.copy()
|
| 190 |
+
|
| 191 |
+
# Expand image tokens
|
| 192 |
+
if image_grid_thw is not None:
|
| 193 |
+
merge_length = self.image_processor.merge_size**2
|
| 194 |
+
placeholder = "]<]placeholder[>["
|
| 195 |
+
index = 0
|
| 196 |
+
for i in range(len(text)):
|
| 197 |
+
while self.IMAGE_TOKEN in text[i]:
|
| 198 |
+
num_tokens = image_grid_thw[index].prod() // merge_length
|
| 199 |
+
text[i] = text[i].replace(
|
| 200 |
+
self.IMAGE_TOKEN,
|
| 201 |
+
self.VISION_START_TOKEN
|
| 202 |
+
+ placeholder * num_tokens
|
| 203 |
+
+ self.VISION_END_TOKEN,
|
| 204 |
+
1,
|
| 205 |
+
)
|
| 206 |
+
index += 1
|
| 207 |
+
text[i] = text[i].replace(placeholder, self.IMAGE_TOKEN)
|
| 208 |
+
|
| 209 |
+
# Expand video tokens
|
| 210 |
+
if video_grid_thw is not None:
|
| 211 |
+
merge_length = self.image_processor.merge_size**2
|
| 212 |
+
placeholder = "]<]placeholder[>["
|
| 213 |
+
index = 0
|
| 214 |
+
for i in range(len(text)):
|
| 215 |
+
while self.VIDEO_TOKEN in text[i]:
|
| 216 |
+
metadata = video_metadata[index]
|
| 217 |
+
grid_t = video_grid_thw[index][0]
|
| 218 |
+
frame_seqlen = video_grid_thw[index][1:].prod() // merge_length
|
| 219 |
+
|
| 220 |
+
video_placeholder = ""
|
| 221 |
+
for frame_idx in range(grid_t):
|
| 222 |
+
if (
|
| 223 |
+
metadata.fps is not None
|
| 224 |
+
and metadata.frames_indices is not None
|
| 225 |
+
):
|
| 226 |
+
ts = (
|
| 227 |
+
metadata.frames_indices[
|
| 228 |
+
min(
|
| 229 |
+
frame_idx
|
| 230 |
+
* self.video_processor.temporal_patch_size,
|
| 231 |
+
len(metadata.frames_indices) - 1,
|
| 232 |
+
)
|
| 233 |
+
]
|
| 234 |
+
/ metadata.fps
|
| 235 |
+
)
|
| 236 |
+
video_placeholder += f"]<]{ts:.1f} seconds[>["
|
| 237 |
+
video_placeholder += (
|
| 238 |
+
self.VISION_START_TOKEN
|
| 239 |
+
+ placeholder * frame_seqlen
|
| 240 |
+
+ self.VISION_END_TOKEN
|
| 241 |
+
)
|
| 242 |
+
|
| 243 |
+
text[i] = text[i].replace(self.VIDEO_TOKEN, video_placeholder, 1)
|
| 244 |
+
index += 1
|
| 245 |
+
text[i] = text[i].replace(placeholder, self.VIDEO_TOKEN)
|
| 246 |
+
|
| 247 |
+
# Tokenize
|
| 248 |
+
return_tensors = output_kwargs["text_kwargs"].pop("return_tensors", None)
|
| 249 |
+
text_inputs = self.tokenizer(text, **output_kwargs["text_kwargs"])
|
| 250 |
+
|
| 251 |
+
return BatchFeature(
|
| 252 |
+
data={**text_inputs, **image_inputs, **video_inputs},
|
| 253 |
+
tensor_type=return_tensors,
|
| 254 |
+
)
|
special_tokens_map.json
ADDED
|
@@ -0,0 +1,16 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"bos_token": {
|
| 3 |
+
"content": "]~b]",
|
| 4 |
+
"lstrip": false,
|
| 5 |
+
"normalized": false,
|
| 6 |
+
"rstrip": false,
|
| 7 |
+
"single_word": false
|
| 8 |
+
},
|
| 9 |
+
"eos_token": {
|
| 10 |
+
"content": "[e~[",
|
| 11 |
+
"lstrip": false,
|
| 12 |
+
"normalized": false,
|
| 13 |
+
"rstrip": false,
|
| 14 |
+
"single_word": false
|
| 15 |
+
}
|
| 16 |
+
}
|
tokenizer.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
tokenizer_config.json
ADDED
|
@@ -0,0 +1,502 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"add_prefix_space": false,
|
| 3 |
+
"added_tokens_decoder": {
|
| 4 |
+
"200000": {
|
| 5 |
+
"content": "]!p~[",
|
| 6 |
+
"lstrip": false,
|
| 7 |
+
"normalized": false,
|
| 8 |
+
"rstrip": false,
|
| 9 |
+
"single_word": false,
|
| 10 |
+
"special": true
|
| 11 |
+
},
|
| 12 |
+
"200001": {
|
| 13 |
+
"content": "<fim_prefix>",
|
| 14 |
+
"lstrip": false,
|
| 15 |
+
"normalized": false,
|
| 16 |
+
"rstrip": false,
|
| 17 |
+
"single_word": false,
|
| 18 |
+
"special": true
|
| 19 |
+
},
|
| 20 |
+
"200002": {
|
| 21 |
+
"content": "<fim_middle>",
|
| 22 |
+
"lstrip": false,
|
| 23 |
+
"normalized": false,
|
| 24 |
+
"rstrip": false,
|
| 25 |
+
"single_word": false,
|
| 26 |
+
"special": true
|
| 27 |
+
},
|
| 28 |
+
"200003": {
|
| 29 |
+
"content": "<fim_suffix>",
|
| 30 |
+
"lstrip": false,
|
| 31 |
+
"normalized": false,
|
| 32 |
+
"rstrip": false,
|
| 33 |
+
"single_word": false,
|
| 34 |
+
"special": true
|
| 35 |
+
},
|
| 36 |
+
"200004": {
|
| 37 |
+
"content": "<fim_pad>",
|
| 38 |
+
"lstrip": false,
|
| 39 |
+
"normalized": false,
|
| 40 |
+
"rstrip": false,
|
| 41 |
+
"single_word": false,
|
| 42 |
+
"special": true
|
| 43 |
+
},
|
| 44 |
+
"200005": {
|
| 45 |
+
"content": "<reponame>",
|
| 46 |
+
"lstrip": false,
|
| 47 |
+
"normalized": false,
|
| 48 |
+
"rstrip": false,
|
| 49 |
+
"single_word": false,
|
| 50 |
+
"special": true
|
| 51 |
+
},
|
| 52 |
+
"200006": {
|
| 53 |
+
"content": "<filename>",
|
| 54 |
+
"lstrip": false,
|
| 55 |
+
"normalized": false,
|
| 56 |
+
"rstrip": false,
|
| 57 |
+
"single_word": false,
|
| 58 |
+
"special": true
|
| 59 |
+
},
|
| 60 |
+
"200007": {
|
| 61 |
+
"content": "<gh_stars>",
|
| 62 |
+
"lstrip": false,
|
| 63 |
+
"normalized": false,
|
| 64 |
+
"rstrip": false,
|
| 65 |
+
"single_word": false,
|
| 66 |
+
"special": true
|
| 67 |
+
},
|
| 68 |
+
"200008": {
|
| 69 |
+
"content": "<issue_start>",
|
| 70 |
+
"lstrip": false,
|
| 71 |
+
"normalized": false,
|
| 72 |
+
"rstrip": false,
|
| 73 |
+
"single_word": false,
|
| 74 |
+
"special": true
|
| 75 |
+
},
|
| 76 |
+
"200009": {
|
| 77 |
+
"content": "<issue_comment>",
|
| 78 |
+
"lstrip": false,
|
| 79 |
+
"normalized": false,
|
| 80 |
+
"rstrip": false,
|
| 81 |
+
"single_word": false,
|
| 82 |
+
"special": true
|
| 83 |
+
},
|
| 84 |
+
"200010": {
|
| 85 |
+
"content": "<issue_closed>",
|
| 86 |
+
"lstrip": false,
|
| 87 |
+
"normalized": false,
|
| 88 |
+
"rstrip": false,
|
| 89 |
+
"single_word": false,
|
| 90 |
+
"special": true
|
| 91 |
+
},
|
| 92 |
+
"200011": {
|
| 93 |
+
"content": "<jupyter_start>",
|
| 94 |
+
"lstrip": false,
|
| 95 |
+
"normalized": false,
|
| 96 |
+
"rstrip": false,
|
| 97 |
+
"single_word": false,
|
| 98 |
+
"special": true
|
| 99 |
+
},
|
| 100 |
+
"200012": {
|
| 101 |
+
"content": "<jupyter_text>",
|
| 102 |
+
"lstrip": false,
|
| 103 |
+
"normalized": false,
|
| 104 |
+
"rstrip": false,
|
| 105 |
+
"single_word": false,
|
| 106 |
+
"special": true
|
| 107 |
+
},
|
| 108 |
+
"200013": {
|
| 109 |
+
"content": "<jupyter_code>",
|
| 110 |
+
"lstrip": false,
|
| 111 |
+
"normalized": false,
|
| 112 |
+
"rstrip": false,
|
| 113 |
+
"single_word": false,
|
| 114 |
+
"special": true
|
| 115 |
+
},
|
| 116 |
+
"200014": {
|
| 117 |
+
"content": "<jupyter_output>",
|
| 118 |
+
"lstrip": false,
|
| 119 |
+
"normalized": false,
|
| 120 |
+
"rstrip": false,
|
| 121 |
+
"single_word": false,
|
| 122 |
+
"special": true
|
| 123 |
+
},
|
| 124 |
+
"200015": {
|
| 125 |
+
"content": "<empty_output>",
|
| 126 |
+
"lstrip": false,
|
| 127 |
+
"normalized": false,
|
| 128 |
+
"rstrip": false,
|
| 129 |
+
"single_word": false,
|
| 130 |
+
"special": true
|
| 131 |
+
},
|
| 132 |
+
"200016": {
|
| 133 |
+
"content": "<commit_before>",
|
| 134 |
+
"lstrip": false,
|
| 135 |
+
"normalized": false,
|
| 136 |
+
"rstrip": false,
|
| 137 |
+
"single_word": false,
|
| 138 |
+
"special": true
|
| 139 |
+
},
|
| 140 |
+
"200017": {
|
| 141 |
+
"content": "<commit_msg>",
|
| 142 |
+
"lstrip": false,
|
| 143 |
+
"normalized": false,
|
| 144 |
+
"rstrip": false,
|
| 145 |
+
"single_word": false,
|
| 146 |
+
"special": true
|
| 147 |
+
},
|
| 148 |
+
"200018": {
|
| 149 |
+
"content": "<commit_after>",
|
| 150 |
+
"lstrip": false,
|
| 151 |
+
"normalized": false,
|
| 152 |
+
"rstrip": false,
|
| 153 |
+
"single_word": false,
|
| 154 |
+
"special": true
|
| 155 |
+
},
|
| 156 |
+
"200019": {
|
| 157 |
+
"content": "]~b]",
|
| 158 |
+
"lstrip": false,
|
| 159 |
+
"normalized": false,
|
| 160 |
+
"rstrip": false,
|
| 161 |
+
"single_word": false,
|
| 162 |
+
"special": true
|
| 163 |
+
},
|
| 164 |
+
"200020": {
|
| 165 |
+
"content": "[e~[",
|
| 166 |
+
"lstrip": false,
|
| 167 |
+
"normalized": false,
|
| 168 |
+
"rstrip": false,
|
| 169 |
+
"single_word": false,
|
| 170 |
+
"special": true
|
| 171 |
+
},
|
| 172 |
+
"200021": {
|
| 173 |
+
"content": "]!d~[",
|
| 174 |
+
"lstrip": false,
|
| 175 |
+
"normalized": false,
|
| 176 |
+
"rstrip": false,
|
| 177 |
+
"single_word": false,
|
| 178 |
+
"special": true
|
| 179 |
+
},
|
| 180 |
+
"200022": {
|
| 181 |
+
"content": "<function_call>",
|
| 182 |
+
"lstrip": false,
|
| 183 |
+
"normalized": false,
|
| 184 |
+
"rstrip": false,
|
| 185 |
+
"single_word": false,
|
| 186 |
+
"special": true
|
| 187 |
+
},
|
| 188 |
+
"200023": {
|
| 189 |
+
"content": "<code_interpreter>",
|
| 190 |
+
"lstrip": false,
|
| 191 |
+
"normalized": false,
|
| 192 |
+
"rstrip": false,
|
| 193 |
+
"single_word": false,
|
| 194 |
+
"special": true
|
| 195 |
+
},
|
| 196 |
+
"200024": {
|
| 197 |
+
"content": "]<]speech[>[",
|
| 198 |
+
"lstrip": false,
|
| 199 |
+
"normalized": false,
|
| 200 |
+
"rstrip": false,
|
| 201 |
+
"single_word": false,
|
| 202 |
+
"special": true
|
| 203 |
+
},
|
| 204 |
+
"200025": {
|
| 205 |
+
"content": "]<]image[>[",
|
| 206 |
+
"lstrip": false,
|
| 207 |
+
"normalized": false,
|
| 208 |
+
"rstrip": false,
|
| 209 |
+
"single_word": false,
|
| 210 |
+
"special": true
|
| 211 |
+
},
|
| 212 |
+
"200026": {
|
| 213 |
+
"content": "]<]video[>[",
|
| 214 |
+
"lstrip": false,
|
| 215 |
+
"normalized": false,
|
| 216 |
+
"rstrip": false,
|
| 217 |
+
"single_word": false,
|
| 218 |
+
"special": true
|
| 219 |
+
},
|
| 220 |
+
"200027": {
|
| 221 |
+
"content": "]<]start of speech[>[",
|
| 222 |
+
"lstrip": false,
|
| 223 |
+
"normalized": false,
|
| 224 |
+
"rstrip": false,
|
| 225 |
+
"single_word": false,
|
| 226 |
+
"special": true
|
| 227 |
+
},
|
| 228 |
+
"200028": {
|
| 229 |
+
"content": "]<]end of speech[>[",
|
| 230 |
+
"lstrip": false,
|
| 231 |
+
"normalized": false,
|
| 232 |
+
"rstrip": false,
|
| 233 |
+
"single_word": false,
|
| 234 |
+
"special": true
|
| 235 |
+
},
|
| 236 |
+
"200029": {
|
| 237 |
+
"content": "]<]start of image[>[",
|
| 238 |
+
"lstrip": false,
|
| 239 |
+
"normalized": false,
|
| 240 |
+
"rstrip": false,
|
| 241 |
+
"single_word": false,
|
| 242 |
+
"special": true
|
| 243 |
+
},
|
| 244 |
+
"200030": {
|
| 245 |
+
"content": "]<]end of image[>[",
|
| 246 |
+
"lstrip": false,
|
| 247 |
+
"normalized": false,
|
| 248 |
+
"rstrip": false,
|
| 249 |
+
"single_word": false,
|
| 250 |
+
"special": true
|
| 251 |
+
},
|
| 252 |
+
"200031": {
|
| 253 |
+
"content": "]<]start of video[>[",
|
| 254 |
+
"lstrip": false,
|
| 255 |
+
"normalized": false,
|
| 256 |
+
"rstrip": false,
|
| 257 |
+
"single_word": false,
|
| 258 |
+
"special": true
|
| 259 |
+
},
|
| 260 |
+
"200032": {
|
| 261 |
+
"content": "]<]end of video[>[",
|
| 262 |
+
"lstrip": false,
|
| 263 |
+
"normalized": false,
|
| 264 |
+
"rstrip": false,
|
| 265 |
+
"single_word": false,
|
| 266 |
+
"special": true
|
| 267 |
+
},
|
| 268 |
+
"200033": {
|
| 269 |
+
"content": "]<]vision pad[>[",
|
| 270 |
+
"lstrip": false,
|
| 271 |
+
"normalized": false,
|
| 272 |
+
"rstrip": false,
|
| 273 |
+
"single_word": false,
|
| 274 |
+
"special": true
|
| 275 |
+
},
|
| 276 |
+
"200034": {
|
| 277 |
+
"content": "]~!b[",
|
| 278 |
+
"lstrip": false,
|
| 279 |
+
"normalized": false,
|
| 280 |
+
"rstrip": false,
|
| 281 |
+
"single_word": false,
|
| 282 |
+
"special": true
|
| 283 |
+
},
|
| 284 |
+
"200035": {
|
| 285 |
+
"content": "<jupyter_error>",
|
| 286 |
+
"lstrip": false,
|
| 287 |
+
"normalized": false,
|
| 288 |
+
"rstrip": false,
|
| 289 |
+
"single_word": false,
|
| 290 |
+
"special": true
|
| 291 |
+
},
|
| 292 |
+
"200036": {
|
| 293 |
+
"content": "<add_file>",
|
| 294 |
+
"lstrip": false,
|
| 295 |
+
"normalized": false,
|
| 296 |
+
"rstrip": false,
|
| 297 |
+
"single_word": false,
|
| 298 |
+
"special": true
|
| 299 |
+
},
|
| 300 |
+
"200037": {
|
| 301 |
+
"content": "<delete_file>",
|
| 302 |
+
"lstrip": false,
|
| 303 |
+
"normalized": false,
|
| 304 |
+
"rstrip": false,
|
| 305 |
+
"single_word": false,
|
| 306 |
+
"special": true
|
| 307 |
+
},
|
| 308 |
+
"200038": {
|
| 309 |
+
"content": "<rename_file>",
|
| 310 |
+
"lstrip": false,
|
| 311 |
+
"normalized": false,
|
| 312 |
+
"rstrip": false,
|
| 313 |
+
"single_word": false,
|
| 314 |
+
"special": true
|
| 315 |
+
},
|
| 316 |
+
"200039": {
|
| 317 |
+
"content": "<edit_file>",
|
| 318 |
+
"lstrip": false,
|
| 319 |
+
"normalized": false,
|
| 320 |
+
"rstrip": false,
|
| 321 |
+
"single_word": false,
|
| 322 |
+
"special": true
|
| 323 |
+
},
|
| 324 |
+
"200040": {
|
| 325 |
+
"content": "<commit_message>",
|
| 326 |
+
"lstrip": false,
|
| 327 |
+
"normalized": false,
|
| 328 |
+
"rstrip": false,
|
| 329 |
+
"single_word": false,
|
| 330 |
+
"special": true
|
| 331 |
+
},
|
| 332 |
+
"200041": {
|
| 333 |
+
"content": "<empty_source_file>",
|
| 334 |
+
"lstrip": false,
|
| 335 |
+
"normalized": false,
|
| 336 |
+
"rstrip": false,
|
| 337 |
+
"single_word": false,
|
| 338 |
+
"special": true
|
| 339 |
+
},
|
| 340 |
+
"200042": {
|
| 341 |
+
"content": "<repo_struct>",
|
| 342 |
+
"lstrip": false,
|
| 343 |
+
"normalized": false,
|
| 344 |
+
"rstrip": false,
|
| 345 |
+
"single_word": false,
|
| 346 |
+
"special": true
|
| 347 |
+
},
|
| 348 |
+
"200043": {
|
| 349 |
+
"content": "<code_context>",
|
| 350 |
+
"lstrip": false,
|
| 351 |
+
"normalized": false,
|
| 352 |
+
"rstrip": false,
|
| 353 |
+
"single_word": false,
|
| 354 |
+
"special": true
|
| 355 |
+
},
|
| 356 |
+
"200044": {
|
| 357 |
+
"content": "<file_content>",
|
| 358 |
+
"lstrip": false,
|
| 359 |
+
"normalized": false,
|
| 360 |
+
"rstrip": false,
|
| 361 |
+
"single_word": false,
|
| 362 |
+
"special": true
|
| 363 |
+
},
|
| 364 |
+
"200045": {
|
| 365 |
+
"content": "<source_files>",
|
| 366 |
+
"lstrip": false,
|
| 367 |
+
"normalized": false,
|
| 368 |
+
"rstrip": false,
|
| 369 |
+
"single_word": false,
|
| 370 |
+
"special": true
|
| 371 |
+
},
|
| 372 |
+
"200046": {
|
| 373 |
+
"content": "<pr_start>",
|
| 374 |
+
"lstrip": false,
|
| 375 |
+
"normalized": false,
|
| 376 |
+
"rstrip": false,
|
| 377 |
+
"single_word": false,
|
| 378 |
+
"special": true
|
| 379 |
+
},
|
| 380 |
+
"200047": {
|
| 381 |
+
"content": "<review_comment>",
|
| 382 |
+
"lstrip": false,
|
| 383 |
+
"normalized": false,
|
| 384 |
+
"rstrip": false,
|
| 385 |
+
"single_word": false,
|
| 386 |
+
"special": true
|
| 387 |
+
},
|
| 388 |
+
"200048": {
|
| 389 |
+
"content": "<filepath>",
|
| 390 |
+
"lstrip": false,
|
| 391 |
+
"normalized": false,
|
| 392 |
+
"rstrip": false,
|
| 393 |
+
"single_word": false,
|
| 394 |
+
"special": true
|
| 395 |
+
},
|
| 396 |
+
"200049": {
|
| 397 |
+
"content": "<file_sep>",
|
| 398 |
+
"lstrip": false,
|
| 399 |
+
"normalized": false,
|
| 400 |
+
"rstrip": false,
|
| 401 |
+
"single_word": false,
|
| 402 |
+
"special": true
|
| 403 |
+
},
|
| 404 |
+
"200050": {
|
| 405 |
+
"content": "<think>",
|
| 406 |
+
"lstrip": false,
|
| 407 |
+
"normalized": false,
|
| 408 |
+
"rstrip": false,
|
| 409 |
+
"single_word": false,
|
| 410 |
+
"special": false
|
| 411 |
+
},
|
| 412 |
+
"200051": {
|
| 413 |
+
"content": "</think>",
|
| 414 |
+
"lstrip": false,
|
| 415 |
+
"normalized": false,
|
| 416 |
+
"rstrip": false,
|
| 417 |
+
"single_word": false,
|
| 418 |
+
"special": false
|
| 419 |
+
},
|
| 420 |
+
"200052": {
|
| 421 |
+
"content": "<tool_call>",
|
| 422 |
+
"lstrip": false,
|
| 423 |
+
"normalized": false,
|
| 424 |
+
"rstrip": false,
|
| 425 |
+
"single_word": false,
|
| 426 |
+
"special": false
|
| 427 |
+
},
|
| 428 |
+
"200053": {
|
| 429 |
+
"content": "</tool_call>",
|
| 430 |
+
"lstrip": false,
|
| 431 |
+
"normalized": false,
|
| 432 |
+
"rstrip": false,
|
| 433 |
+
"single_word": false,
|
| 434 |
+
"special": false
|
| 435 |
+
},
|
| 436 |
+
"200054": {
|
| 437 |
+
"content": "]<]frame[>[",
|
| 438 |
+
"lstrip": false,
|
| 439 |
+
"normalized": false,
|
| 440 |
+
"rstrip": false,
|
| 441 |
+
"single_word": false,
|
| 442 |
+
"special": true
|
| 443 |
+
},
|
| 444 |
+
"200055": {
|
| 445 |
+
"content": "]<]start of frame[>[",
|
| 446 |
+
"lstrip": false,
|
| 447 |
+
"normalized": false,
|
| 448 |
+
"rstrip": false,
|
| 449 |
+
"single_word": false,
|
| 450 |
+
"special": true
|
| 451 |
+
},
|
| 452 |
+
"200056": {
|
| 453 |
+
"content": "]<]end of frame[>[",
|
| 454 |
+
"lstrip": false,
|
| 455 |
+
"normalized": false,
|
| 456 |
+
"rstrip": false,
|
| 457 |
+
"single_word": false,
|
| 458 |
+
"special": true
|
| 459 |
+
},
|
| 460 |
+
"200057": {
|
| 461 |
+
"content": "<|content_altered_placeholder|>",
|
| 462 |
+
"lstrip": false,
|
| 463 |
+
"normalized": false,
|
| 464 |
+
"rstrip": false,
|
| 465 |
+
"single_word": false,
|
| 466 |
+
"special": true
|
| 467 |
+
},
|
| 468 |
+
"200058": {
|
| 469 |
+
"content": "]<]minimax[>[",
|
| 470 |
+
"lstrip": false,
|
| 471 |
+
"normalized": false,
|
| 472 |
+
"rstrip": false,
|
| 473 |
+
"single_word": false,
|
| 474 |
+
"special": false
|
| 475 |
+
},
|
| 476 |
+
"200059": {
|
| 477 |
+
"content": "<mm:think>",
|
| 478 |
+
"lstrip": false,
|
| 479 |
+
"normalized": false,
|
| 480 |
+
"rstrip": false,
|
| 481 |
+
"single_word": false,
|
| 482 |
+
"special": false
|
| 483 |
+
},
|
| 484 |
+
"200060": {
|
| 485 |
+
"content": "</mm:think>",
|
| 486 |
+
"lstrip": false,
|
| 487 |
+
"normalized": false,
|
| 488 |
+
"rstrip": false,
|
| 489 |
+
"single_word": false,
|
| 490 |
+
"special": false
|
| 491 |
+
}
|
| 492 |
+
},
|
| 493 |
+
"bos_token": "]~b]",
|
| 494 |
+
"clean_up_tokenization_spaces": false,
|
| 495 |
+
"eos_token": "[e~[",
|
| 496 |
+
"pad_token": "]!p~[",
|
| 497 |
+
"model_max_length": 40960000,
|
| 498 |
+
"tokenizer_class": "PreTrainedTokenizerFast",
|
| 499 |
+
"unk_token": "[e~[",
|
| 500 |
+
"chat_template": "{# ---------- special token variables ---------- #}\n{%- set ns_token = ']<]minimax[>[' -%}\n{%- set bod_token = ']~!b[' -%}\n{%- set bos_token = ']~b]' -%}\n{%- set eos_token = '[e~[' -%}\n{%- set toolcall_begin_token = ns_token ~ '<tool_call>' -%}\n{%- set toolcall_end_token = ns_token ~ '</tool_call>' -%}\n{%- set think_begin_token = '<mm:think>' -%}\n{%- set think_end_token = '</mm:think>' -%}\n{%- set image_token = ']<]image[>[' -%}\n{%- set video_token = ']<]video[>[' -%}\n{#- Thinking mode: \"enabled\" / \"disabled\" / \"adaptive\" / not defined -#}\n{#- Recursive XML renderer for tool_call arguments ======================== -#}\n{#- None values are intentionally skipped in mapping iteration so that\n `<key>null</key>` (which would round-trip to the literal string \"null\")\n never appears in the rendered tool_call. The convention is: omit the\n field entirely. The top-level `_args` loop applies the same rule.\n The `val is none` branch below is a safety net only — upstream cleaning\n (drop_none_in_tool_arguments) should ensure no None ever reaches here. -#}\n{%- macro to_xml(val, ns) -%}\n{%- if val is mapping -%}\n{%- for k, v in val.items() if v is not none -%}\n{{ ns }}<{{ k }}>{{ to_xml(v, ns) }}{{ ns }}</{{ k }}>\n{%- endfor -%}\n{%- elif val is iterable and val is not string -%}\n{%- for item in val -%}\n{{ ns }}<item>{{ to_xml(item, ns) }}{{ ns }}</item>\n{%- endfor -%}\n{%- elif val is none -%}\n{#- Should be unreachable when upstream cleaning is applied. -#}\n{%- elif val is boolean -%}\n{{ val | tojson }}\n{%- else -%}\n{{ val }}\n{%- endif -%}\n{%- endmacro -%}\n{#- Tool Rendering Functions ============================================== -#}\n{%- macro render_tool_namespace(namespace_name, tool_list) -%}\n{%- for tool in tool_list -%}\n<tool>{{ tool.function | tojson(ensure_ascii=False) }}</tool>\n{% endfor -%}\n{%- endmacro -%}\n{%- macro visible_text(content) -%}\n {%- if content is string -%}\n {{ content }}\n {%- elif content is iterable and content is not mapping -%}\n {%- for item in content -%}\n {%- if item is mapping and item.type == 'text' -%}\n {{- item.text }}\n {%- elif item is mapping and item.type == 'image' -%}\n {{- image_token }}\n {%- elif item is mapping and item.type == 'video' -%}\n {{- video_token}}\n {%- elif item is string -%}\n {{- item }}\n {%- endif -%}\n {%- endfor -%}\n {%- elif content is none -%}\n {{- '' }}\n {%- else -%}\n {{- content }}\n {%- endif -%}\n{%- endmacro -%}\n{#- System Message Construction ============================================ -#}\n{%- macro build_system_message(system_message) -%}\n {%- if system_message and system_message.content -%}\n {{- visible_text(system_message.content) }}\n {#- Handle current_date -#}\n {%- if system_message.current_date -%}\n {{- '\\n' ~ 'Current date: ' + system_message.current_date }}\n {%- endif -%}\n {#- Handle current_location -#}\n {%- if system_message.current_location -%}\n {{- '\\n' ~ 'Current location: ' + system_message.current_location }}\n {%- endif -%}\n {%- endif -%}\n\n {#- Thinking mode instructions -#}\n {{- '\\n\\n<thinking_instructions>\\n' }}\n {{- 'You have a thinking capability that allows you to reason step by step before responding. When thinking is enabled, wrap your reasoning in ' ~ think_begin_token ~ think_end_token ~ ' tags before your response. When thinking is disabled, begin your response directly after the ' ~ think_end_token ~ ' prefix. When thinking is adaptive, decide on your own whether to think for the current turn.\\n' }}\n {%- if thinking_mode is defined -%}\n {%- if thinking_mode == \"enabled\" -%}\n {{- 'Current thinking mode: enabled. You MUST think step by step before every response, including after receiving function/tool results.\\n' }}\n {%- elif thinking_mode == \"disabled\" -%}\n {{- 'Current thinking mode: disabled. Do not output any thinking process.\\n' }}\n {%- elif thinking_mode == \"adaptive\" -%}\n {{- 'Current thinking mode: adaptive. You are encouraged to think for complex decision-making, multi-step reasoning, or when analyzing function/tool results.\\n' }}\n {%- endif -%}\n {%- else -%}\n {{- 'Current thinking mode: adaptive. You are encouraged to think for complex decision-making, multi-step reasoning, or when analyzing function/tool results.\\n' }}\n {%- endif -%}\n {{- '</thinking_instructions>' }}\n {%- if template_model_id is defined -%}\n {{- '\\n\\n<model_id>' ~ template_model_id ~ '</model_id>' }}\n {%- endif -%}\n{%- endmacro -%}\n{%- macro build_developer_message(developer_message) -%}\n {%- if developer_message and developer_message.content -%}\n {{- visible_text(developer_message.content) }}\n {%- else -%}\n {%- if model_identity is not defined -%}\n {%- set model_identity = \"You are a helpful assistant.\" -%}\n {%- endif -%}\n {{- model_identity }}\n {%- endif -%}\n{%- endmacro -%}\n{#- Main Template Logic ================================================= -#}\n{#- Role mapping: root -> system sp (high priority), system/developer -> developer sp (low priority) -#}\n{%- set system_message = none -%}\n{%- set developer_message = none -%}\n{%- set conversation_messages = messages -%}\n{%- if messages and messages[0].role == \"root\" -%}\n {%- set system_message = messages[0] -%}\n {%- set conversation_messages = messages[1:] -%}\n {%- if conversation_messages and conversation_messages[0].role in [\"system\", \"developer\"] -%}\n {%- set developer_message = conversation_messages[0] -%}\n {%- set conversation_messages = conversation_messages[1:] -%}\n {%- endif -%}\n{%- elif messages and messages[0].role in [\"system\", \"developer\"] -%}\n {%- set developer_message = messages[0] -%}\n {%- set conversation_messages = messages[1:] -%}\n{%- endif -%}\n{#- Render system sp (higher priority, root role only) -#}\n{{- bod_token ~ bos_token ~ 'system' ~ '\\n' }}\n{{- build_system_message(system_message) }}\n{{- eos_token ~ '\\n' }}\n\n{#- Render developer sp (lower priority: system/developer role + tools) -#}\n{{- bos_token ~ 'developer' ~ '\\n' }}\n{{- build_developer_message(developer_message) }}\n{%- if tools -%}\n {{- '\\n\\n' ~ '# Tools' ~ '\\n' ~ 'You may call one or more tools to assist with the user query.\\nHere are the tools available in JSONSchema format:' ~ '\\n' }}\n {{- '\\n' ~ '<tools>' ~ '\\n' }}\n {{- render_tool_namespace(\"functions\", tools) }}\n {{- '</tools>' ~ '\\n\\n' }}\n {{- 'To call tools, wrap all invocations in a single ' ~ toolcall_begin_token ~ toolcall_end_token ~ ' block. Parameter values containing nested objects or arrays are recursively expanded into XML elements. Example:\\n' }}\n {{- '\\n' ~ toolcall_begin_token ~ '\\n' }}\n {{- ns_token + '<invoke name=\"tool-name-1\">' }}\n {{- ns_token + '<param-1>value-1' + ns_token + '</param-1>' }}\n {{- ns_token + '<param-2>' }}\n {{- ns_token + '<item>' }}\n {{- ns_token + '<key-a>val-a' + ns_token + '</key-a>' }}\n {{- ns_token + '<key-b>val-b' + ns_token + '</key-b>' }}\n {{- ns_token + '</item>' }}\n {{- ns_token + '</param-2>' }}\n {{- ns_token + '</invoke>\\n' }}\n {{- ns_token + '<invoke name=\"tool-name-2\">' }}\n {{- ns_token + '<param-1>value-1' + ns_token + '</param-1>' }}\n {{- ns_token + '</invoke>\\n' }}\n {{- toolcall_end_token }}\n{%- endif -%}\n{{- eos_token ~ '\\n' }}\n\n{#- Render messages -#}\n{%- set last_tool_call = namespace(name=none) -%}\n{%- for message in conversation_messages -%}\n {%- if message.role == 'assistant' -%}\n {{- bos_token ~ 'ai' ~ '\\n' }}\n\n {%- set reasoning_content = '' %}\n {%- set content = visible_text(message.content) %}\n {%- if message.reasoning_content is string %}\n {%- set reasoning_content = message.reasoning_content %}\n {%- else %}\n {%- if think_end_token in content %}\n {%- set reasoning_content = content.split(think_end_token)[0].strip('\\n').split(think_begin_token)[-1].strip('\\n') %}\n {%- set content = content.split(think_end_token)[-1].strip('\\n') %}\n {%- endif %}\n {%- endif %}\n\n {%- if reasoning_content -%}\n {#- Render thinking for every assistant turn (all-turn visible) -#}\n {{- think_begin_token ~ reasoning_content ~ think_end_token }}\n {%- else -%}\n {#- No thinking rendered → prefix with think_end_token -#}\n {{- think_end_token }}\n {%- endif -%}\n\n {%- if content -%}\n {{- content }}\n {%- endif -%}\n {%- if message.tool_calls -%}\n {{- toolcall_begin_token ~ '\\n' }}\n\n {%- for tool_call in message.tool_calls -%}\n {%- if tool_call.function -%}\n {%- set tool_call = tool_call.function -%}\n {%- endif -%}\n{{- ns_token + '<invoke name=\"' + tool_call.name + '\">' }}\n{%- set _args = tool_call.arguments -%}\n{%- for k, v in _args.items() if v is not none %}\n{{- ns_token + '<' + k + '>' -}}\n{{- to_xml(v, ns_token) -}}\n{{- ns_token + '</' + k + '>' }}\n{%- endfor -%}\n{{- ns_token + '</invoke>' ~ '\\n' }}\n {%- endfor -%}\n\n {{- toolcall_end_token }}\n {%- if message.tool_calls[-1].function -%}\n {%- set last_tool_call.name = message.tool_calls[-1].function.name -%}\n {%- else -%}\n {%- set last_tool_call.name = message.tool_calls[-1].name -%}\n {%- endif -%}\n {%- else -%}\n {%- set last_tool_call.name = none -%}\n {%- endif -%}\n {{- eos_token ~ '\\n' }}\n\n {%- elif message.role == 'tool' -%}\n {%- if last_tool_call.name is none -%}\n {{- raise_exception(\"Message has tool role, but there was no previous assistant message with a tool call!\") }}\n {%- endif -%}\n {%- if loop.first or (conversation_messages[loop.index0 - 1].role != 'tool') -%}\n {{- bos_token ~ 'tool' }}\n {%- endif -%}\n {{- '\\n<response>' }}\n {%- if message.content is string -%}\n {{- message.content }}\n {%- else -%}\n {%- for tr in message.content -%}\n {%- if tr is mapping and tr.type is defined and tr.type == 'image' -%}\n {{- image_token }}\n {%- elif tr is mapping and tr.type is defined and tr.type == 'video' -%}\n {{- video_token }}\n {%- else -%}\n {{- tr.output if tr.output is defined else (tr.text if tr.type == 'text' and tr.text is defined else tr) }}\n {%- endif -%}\n {%- endfor -%}\n {%- endif -%}\n {{- '</response>' }}\n {%- if loop.last or (conversation_messages[loop.index0 + 1].role != 'tool') -%}\n {{- eos_token ~ '\\n' -}}\n {%- endif -%}\n\n {%- elif message.role == 'user' -%}\n {{- bos_token ~ 'user' ~ '\\n' }}\n {{- visible_text(message.content) }}\n {{- eos_token ~ '\\n' }}\n {%- endif -%}\n{%- endfor -%}\n\n{#- Generation prompt -#}\n{%- if add_generation_prompt -%}\n{{- bos_token ~ 'ai' ~ '\\n' }}\n{%- if thinking_mode is defined and thinking_mode == \"disabled\" -%}\n {{- think_end_token }}\n{%- elif thinking_mode is defined and thinking_mode == \"adaptive\" -%}\n {#- adaptive: no prefix, let model decide -#}\n{%- elif thinking_mode is defined and thinking_mode == \"enabled\" -%}\n {#- enabled or not defined: default to think -#}\n {{- think_begin_token }}\n{%- else -%}\n {#- adaptive: no prefix, let model decide -#}\n{%- endif -%}\n{%- endif -%}\n",
|
| 501 |
+
"chat_template_file": "chat_template.jinja"
|
| 502 |
+
}
|
video_processor.py
ADDED
|
@@ -0,0 +1,208 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Copyright 2023-2024 SGLang Team
|
| 2 |
+
# Licensed under the Apache License, Version 2.0 (the "License");
|
| 3 |
+
"""
|
| 4 |
+
MiniMax VL family HuggingFace-compatible VideoProcessor.
|
| 5 |
+
"""
|
| 6 |
+
|
| 7 |
+
import math
|
| 8 |
+
from typing import List, Optional, Tuple, Union
|
| 9 |
+
|
| 10 |
+
import torch
|
| 11 |
+
import torchvision
|
| 12 |
+
from torchvision.transforms import InterpolationMode
|
| 13 |
+
from transformers import BatchFeature
|
| 14 |
+
from transformers.image_utils import PILImageResampling, SizeDict
|
| 15 |
+
from transformers.processing_utils import (
|
| 16 |
+
Unpack,
|
| 17 |
+
VideosKwargs,
|
| 18 |
+
)
|
| 19 |
+
from transformers.utils import TensorType
|
| 20 |
+
from transformers.video_processing_utils import BaseVideoProcessor
|
| 21 |
+
from transformers.video_utils import group_videos_by_shape, reorder_videos
|
| 22 |
+
|
| 23 |
+
MAX_RATIO = 200
|
| 24 |
+
|
| 25 |
+
|
| 26 |
+
def round_by_factor(number: int, factor: int) -> int:
|
| 27 |
+
return round(number / factor) * factor
|
| 28 |
+
|
| 29 |
+
|
| 30 |
+
def ceil_by_factor(number: int, factor: int) -> int:
|
| 31 |
+
return math.ceil(number / factor) * factor
|
| 32 |
+
|
| 33 |
+
|
| 34 |
+
def floor_by_factor(number: int, factor: int) -> int:
|
| 35 |
+
return math.floor(number / factor) * factor
|
| 36 |
+
|
| 37 |
+
|
| 38 |
+
def smart_resize(
|
| 39 |
+
height: int,
|
| 40 |
+
width: int,
|
| 41 |
+
factor: int = 28,
|
| 42 |
+
min_pixels: int = 4 * 28 * 28,
|
| 43 |
+
max_pixels: int = 451584,
|
| 44 |
+
) -> tuple[int, int]:
|
| 45 |
+
if max(height, width) / min(height, width) > MAX_RATIO:
|
| 46 |
+
raise ValueError(
|
| 47 |
+
f"absolute aspect ratio must be smaller than {MAX_RATIO}, "
|
| 48 |
+
f"got {max(height, width) / min(height, width)}"
|
| 49 |
+
)
|
| 50 |
+
h_bar = max(factor, round_by_factor(height, factor))
|
| 51 |
+
w_bar = max(factor, round_by_factor(width, factor))
|
| 52 |
+
if h_bar * w_bar > max_pixels:
|
| 53 |
+
beta = math.sqrt((height * width) / max_pixels)
|
| 54 |
+
h_bar = floor_by_factor(height / beta, factor)
|
| 55 |
+
w_bar = floor_by_factor(width / beta, factor)
|
| 56 |
+
elif h_bar * w_bar < min_pixels:
|
| 57 |
+
beta = math.sqrt(min_pixels / (height * width))
|
| 58 |
+
h_bar = ceil_by_factor(height * beta, factor)
|
| 59 |
+
w_bar = ceil_by_factor(width * beta, factor)
|
| 60 |
+
return h_bar, w_bar
|
| 61 |
+
|
| 62 |
+
|
| 63 |
+
class MiniMaxM3VLVideoProcessorKwargs(VideosKwargs, total=False):
|
| 64 |
+
patch_size: int
|
| 65 |
+
temporal_patch_size: int
|
| 66 |
+
merge_size: int
|
| 67 |
+
min_pixels: int
|
| 68 |
+
max_pixels: int
|
| 69 |
+
total_pixels: int
|
| 70 |
+
min_frames: int
|
| 71 |
+
max_frames: int
|
| 72 |
+
fps: float | int
|
| 73 |
+
|
| 74 |
+
|
| 75 |
+
class MiniMaxM3VLVideoProcessor(BaseVideoProcessor):
|
| 76 |
+
do_resize = True
|
| 77 |
+
resample = PILImageResampling.BICUBIC
|
| 78 |
+
size = {"height": 672, "width": 672}
|
| 79 |
+
default_to_square = False
|
| 80 |
+
do_rescale = True
|
| 81 |
+
rescale_factor = 1 / 255
|
| 82 |
+
do_normalize = True
|
| 83 |
+
image_mean = [0.48145466, 0.4578275, 0.40821073]
|
| 84 |
+
image_std = [0.26862954, 0.26130258, 0.27577711]
|
| 85 |
+
do_convert_rgb = True
|
| 86 |
+
do_sample_frames = False
|
| 87 |
+
patch_size = 14
|
| 88 |
+
temporal_patch_size = 2
|
| 89 |
+
merge_size = 2
|
| 90 |
+
min_pixels = 4 * 28 * 28
|
| 91 |
+
max_pixels = 768 * 28 * 28 # 602,112
|
| 92 |
+
total_pixels = int(64000 * 28 * 28 * 0.9) # ~45M, ~64k tokens budget
|
| 93 |
+
fps = 1.0
|
| 94 |
+
min_frames = 4
|
| 95 |
+
max_frames = 768
|
| 96 |
+
valid_kwargs = MiniMaxM3VLVideoProcessorKwargs
|
| 97 |
+
model_input_names = ["pixel_values_videos", "video_grid_thw"]
|
| 98 |
+
|
| 99 |
+
def __init__(self, **kwargs: Unpack[MiniMaxM3VLVideoProcessorKwargs]):
|
| 100 |
+
super().__init__(**kwargs)
|
| 101 |
+
|
| 102 |
+
def _preprocess(
|
| 103 |
+
self,
|
| 104 |
+
videos: List[torch.Tensor],
|
| 105 |
+
do_convert_rgb: bool,
|
| 106 |
+
do_resize: bool,
|
| 107 |
+
size: SizeDict,
|
| 108 |
+
resample: PILImageResampling | InterpolationMode | int | None,
|
| 109 |
+
do_rescale: bool,
|
| 110 |
+
rescale_factor: float,
|
| 111 |
+
do_normalize: bool,
|
| 112 |
+
image_mean: float | List[float] | None,
|
| 113 |
+
image_std: float | List[float] | None,
|
| 114 |
+
patch_size: int,
|
| 115 |
+
temporal_patch_size: int,
|
| 116 |
+
merge_size: int,
|
| 117 |
+
min_pixels: int,
|
| 118 |
+
max_pixels: int,
|
| 119 |
+
return_tensors: str | TensorType | None = None,
|
| 120 |
+
**kwargs,
|
| 121 |
+
) -> BatchFeature:
|
| 122 |
+
grouped_videos, grouped_videos_index = group_videos_by_shape(videos)
|
| 123 |
+
resized_videos_grouped = {}
|
| 124 |
+
factor = patch_size * merge_size
|
| 125 |
+
for shape, stacked_videos in grouped_videos.items():
|
| 126 |
+
batch_size, num_frames, channels, height, width = stacked_videos.shape
|
| 127 |
+
resized_height, resized_width = height, width
|
| 128 |
+
if do_resize:
|
| 129 |
+
resized_height, resized_width = smart_resize(
|
| 130 |
+
height, width, factor=factor,
|
| 131 |
+
min_pixels=min_pixels, max_pixels=max_pixels,
|
| 132 |
+
)
|
| 133 |
+
stacked_videos = stacked_videos.view(
|
| 134 |
+
batch_size * num_frames, channels, height, width
|
| 135 |
+
)
|
| 136 |
+
stacked_videos = self.resize(
|
| 137 |
+
stacked_videos,
|
| 138 |
+
size=SizeDict(height=resized_height, width=resized_width),
|
| 139 |
+
resample=resample,
|
| 140 |
+
)
|
| 141 |
+
stacked_videos = stacked_videos.view(
|
| 142 |
+
batch_size,
|
| 143 |
+
num_frames,
|
| 144 |
+
channels,
|
| 145 |
+
resized_height,
|
| 146 |
+
resized_width,
|
| 147 |
+
)
|
| 148 |
+
resized_videos_grouped[shape] = stacked_videos
|
| 149 |
+
resized_videos = reorder_videos(resized_videos_grouped, grouped_videos_index)
|
| 150 |
+
|
| 151 |
+
grouped_videos, grouped_videos_index = group_videos_by_shape(resized_videos)
|
| 152 |
+
processed_videos_grouped = {}
|
| 153 |
+
processed_grids = {}
|
| 154 |
+
for shape, stacked_videos in grouped_videos.items():
|
| 155 |
+
resized_height, resized_width = stacked_videos.shape[-2:]
|
| 156 |
+
patches = self.rescale_and_normalize(
|
| 157 |
+
stacked_videos,
|
| 158 |
+
do_rescale,
|
| 159 |
+
rescale_factor,
|
| 160 |
+
do_normalize,
|
| 161 |
+
image_mean,
|
| 162 |
+
image_std,
|
| 163 |
+
)
|
| 164 |
+
|
| 165 |
+
if pad := -patches.shape[1] % temporal_patch_size:
|
| 166 |
+
repeats = patches[:, -1:].expand(-1, pad, -1, -1, -1)
|
| 167 |
+
patches = torch.cat([patches, repeats], dim=1)
|
| 168 |
+
|
| 169 |
+
batch_size, grid_t, channels = patches.shape[:3]
|
| 170 |
+
grid_t = grid_t // temporal_patch_size
|
| 171 |
+
grid_h, grid_w = resized_height // patch_size, resized_width // patch_size
|
| 172 |
+
|
| 173 |
+
patches = patches.view(
|
| 174 |
+
batch_size,
|
| 175 |
+
grid_t,
|
| 176 |
+
temporal_patch_size,
|
| 177 |
+
channels,
|
| 178 |
+
grid_h // merge_size,
|
| 179 |
+
merge_size,
|
| 180 |
+
patch_size,
|
| 181 |
+
grid_w // merge_size,
|
| 182 |
+
merge_size,
|
| 183 |
+
patch_size,
|
| 184 |
+
)
|
| 185 |
+
patches = patches.permute(0, 1, 4, 7, 5, 8, 3, 2, 6, 9)
|
| 186 |
+
flatten_patches = patches.reshape(
|
| 187 |
+
batch_size,
|
| 188 |
+
grid_t * grid_h * grid_w,
|
| 189 |
+
channels * temporal_patch_size * patch_size * patch_size,
|
| 190 |
+
)
|
| 191 |
+
|
| 192 |
+
processed_videos_grouped[shape] = flatten_patches
|
| 193 |
+
processed_grids[shape] = [[grid_t, grid_h, grid_w]] * batch_size
|
| 194 |
+
|
| 195 |
+
processed_videos = reorder_videos(
|
| 196 |
+
processed_videos_grouped, grouped_videos_index
|
| 197 |
+
)
|
| 198 |
+
processed_grids = reorder_videos(processed_grids, grouped_videos_index)
|
| 199 |
+
pixel_values_videos = torch.cat(processed_videos, dim=0)
|
| 200 |
+
video_grid_thw = torch.tensor(processed_grids, dtype=torch.long)
|
| 201 |
+
|
| 202 |
+
return BatchFeature(
|
| 203 |
+
data={
|
| 204 |
+
"pixel_values_videos": pixel_values_videos,
|
| 205 |
+
"video_grid_thw": video_grid_thw,
|
| 206 |
+
},
|
| 207 |
+
tensor_type=return_tensors,
|
| 208 |
+
)
|
vocab.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|