初始化项目，由ModelHub XC社区提供模型

Model: AIDC-AI/Ovis2.5-9B
Source: Original Platform

.gitattributes

@@ -0,0 +1,51 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bin.* filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zstandard filter=lfs diff=lfs merge=lfs -text
+*.tfevents* filter=lfs diff=lfs merge=lfs -text
+*.db* filter=lfs diff=lfs merge=lfs -text
+*.ark* filter=lfs diff=lfs merge=lfs -text
+**/*ckpt*data* filter=lfs diff=lfs merge=lfs -text
+**/*ckpt*.meta filter=lfs diff=lfs merge=lfs -text
+**/*ckpt*.index filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.gguf* filter=lfs diff=lfs merge=lfs -text
+*.ggml filter=lfs diff=lfs merge=lfs -text
+*.llamafile* filter=lfs diff=lfs merge=lfs -text
+*.pt2 filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.tar filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text
+
+merges.txt filter=lfs diff=lfs merge=lfs -text
+tokenizer.json filter=lfs diff=lfs merge=lfs -text
+vocab.json filter=lfs diff=lfs merge=lfs -text

LICENSE

@@ -0,0 +1,10 @@
+Copyright (C) 2025 AIDC-AI
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+    http://www.apache.org/licenses/LICENSE-2.0
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.

NOTICE

@@ -0,0 +1,5 @@
+Copyright (C) 2025 AIDC-AI
+Licensed under the Apache 2.0 (the "License").
+The model was trained based on the following models:
+1. Qwen3-8B (https://huggingface.co/Qwen/Qwen3-8B), license: Apache License 2.0 (https://huggingface.co/Qwen/Qwen3-8B/blob/main/LICENSE). 
+2. Siglip2 (https://huggingface.co/google/siglip2-so400m-patch16-512), license: Apache License 2.0 (https://choosealicense.com/licenses/apache-2.0/). 

README.md

@@ -0,0 +1,339 @@
+---
+license: apache-2.0
+datasets:
+- AIDC-AI/Ovis-dataset
+library_name: transformers
+tags:
+- MLLM
+pipeline_tag: image-text-to-text
+language:
+- en
+- zh
+---
+# Ovis2.5-9B
+<div align="center">
+  <img src=https://cdn-uploads.huggingface.co/production/uploads/637aebed7ce76c3b834cea37/3IK823BZ8w-mz_QfeYkDn.png width="30%"/>
+</div>
+
+<p align="center">
+  <a href="https://arxiv.org/abs/2508.11737"><img src="https://img.shields.io/badge/📖_Technical_Report-Ovis2.5-b31b1b.svg" alt="technical report"></a>
+  <a href="https://github.com/AIDC-AI/Ovis"><img src="https://img.shields.io/badge/GitHub-AIDC--AI/Ovis-blue?style=flat&logo=github" alt="code"></a>
+  <a href="https://huggingface.co/spaces/AIDC-AI/Ovis2.5-9B"><img src="https://img.shields.io/badge/🎨_HF_Spaces-AIDC--AI/Ovis2.5--9B-lightblack" alt="demo"></a>
+  <a href="https://huggingface.co/collections/AIDC-AI/ovis25-689ec1474633b2aab8809335"><img src="https://img.shields.io/badge/🤗_Models-AIDC--AI/Ovis2.5-yellow" alt="models"></a>
+</p>
+
+## Introduction
+
+
+We are pleased to announce the release of **Ovis2.5**, the successor to Ovis2, designed for native-resolution visual perception and enhanced multimodal reasoning. 
+It integrates a native-resolution vision transformer (NaViT) that processes images at their original, variable resolutions, eliminating the need for fixed-resolution tiling and preserving both fine details and global layout—crucial for visually dense content such as charts and diagrams. 
+To strengthen reasoning, Ovis2.5 is trained not only on linear chain-of-thought (CoT) but also on reflective reasoning, including self-checking and revision. 
+This advanced capability is available at inference as an optional *thinking mode*, enabling users to trade latency for higher accuracy on complex inputs.
+
+Building on these advances, **Ovis2.5-9B** achieves an average score of 78.3 on the OpenCompass multimodal evaluation suite (SOTA among open-source MLLMs under 40B parameters), while the lightweight **Ovis2.5-2B** scores 73.9, continuing the “small model, big performance” philosophy for resource-constrained scenarios.
+
+
+<div align="center">
+    <img src="https://cdn-uploads.huggingface.co/production/uploads/637aebed7ce76c3b834cea37/kh-1dhZRAduP-P4SkIhXr.png" width="100%" />
+</div>
+
+**Key Features**  
+* **Native-Resolution Perception** — NaViT vision encoder preserves fine details and global structure without lossy tiling.  
+* **Deep-Reasoning Capability** — Optional *thinking mode* for self-checking and revision beyond linear CoT. *Thinking budget* supported.
+* **Chart & Document OCR** — State-of-the-art at its scale for complex chart analysis, document understanding (including tables and forms), and OCR.  
+* **Broad Task Coverage** — Demonstrates leading performance on image reasoning, video understanding, and grounding benchmarks, showcasing strong general multimodal capability.
+
+<div align="center">
+    <img src="https://cdn-uploads.huggingface.co/production/uploads/637aebed7ce76c3b834cea37/4kw2RRUhXDiMZdU7wGOfP.png" width="100%" />
+</div>
+
+## Quick Inference
+
+Below is a simple example demonstrating how to run Ovis2.5 with a single image input. For accelerated inference with **vLLM**, refer to [GitHub](https://github.com/AIDC-AI/Ovis).
+
+First, install the required dependencies:
+
+```bash
+pip install torch==2.4.0 transformers==4.51.3 numpy==1.25.0 pillow==10.3.0 moviepy==1.0.3
+pip install flash-attn==2.7.0.post2 --no-build-isolation
+```
+
+Then, run the following code.
+
+```python
+import torch
+import requests
+from PIL import Image
+from transformers import AutoModelForCausalLM
+
+MODEL_PATH = "AIDC-AI/Ovis2.5-9B"
+
+# Thinking mode & budget
+enable_thinking = True
+enable_thinking_budget = True  # Only effective if enable_thinking is True.
+
+# Total tokens for thinking + answer. Ensure: max_new_tokens > thinking_budget + 25
+max_new_tokens = 3072
+thinking_budget = 2048
+
+model = AutoModelForCausalLM.from_pretrained(
+    MODEL_PATH,
+    torch_dtype=torch.bfloat16,
+    trust_remote_code=True
+).cuda()
+
+messages = [{
+    "role": "user",
+    "content": [
+        {"type": "image", "image": Image.open(requests.get("https://cdn-uploads.huggingface.co/production/uploads/658a8a837959448ef5500ce5/TIlymOb86R6_Mez3bpmcB.png", stream=True).raw)},
+        {"type": "text", "text": "Calculate the sum of the numbers in the middle box in figure (c)."},
+    ],
+}]
+
+input_ids, pixel_values, grid_thws = model.preprocess_inputs(
+    messages=messages,
+    add_generation_prompt=True,
+    enable_thinking=enable_thinking
+)
+input_ids = input_ids.cuda()
+pixel_values = pixel_values.cuda() if pixel_values is not None else None
+grid_thws = grid_thws.cuda() if grid_thws is not None else None
+
+outputs = model.generate(
+    inputs=input_ids,
+    pixel_values=pixel_values,
+    grid_thws=grid_thws,
+    enable_thinking=enable_thinking,
+    enable_thinking_budget=enable_thinking_budget,
+    max_new_tokens=max_new_tokens,
+    thinking_budget=thinking_budget,
+)
+
+response = model.text_tokenizer.decode(outputs[0], skip_special_tokens=True)
+print(response)
+```
+
+The thinking and thinking budget logic can be applied in the same way for multi-image, video and pure text scenarios.
+
+**Note (answer extraction for CoT/Thinking):**
+To make evaluation and usage easier, we recommend appending a fixed suffix to prompts when using chain-of-thought (CoT) or thinking mode. This ensures the model clearly outputs a final answer that can be extracted programmatically:
+
+```
+End your response with 'Final answer: '.
+```
+
+For example:
+
+```
+Calculate the sum of the numbers in the middle box in figure (c).
+End your response with 'Final answer: '.
+```
+
+
+**Tip:** The sections below include an optional streaming helper (compatible with two-phase thinking/budget runs) and extra inference modes: multi-image, video, and text-only.
+
+<details>
+<summary>Optional: Streaming (Advanced)</summary>
+
+To support thinking budget, we modified the implementation of the Ovis `generate` method and the default `TextIteratorStreamer` is now incompatible. If you need to stream model output, be sure to use the helper class below.
+
+```python
+# --- Budget-aware streamer helper ---
+from transformers import TextIteratorStreamer
+
+class BudgetAwareTextStreamer(TextIteratorStreamer):
+    """A streamer compatible with Ovis two-phase generation.
+
+    Call .manual_end() after generation to flush any remaining text.
+    """
+    def manual_end(self):
+        if len(self.token_cache) > 0:
+            text = self.tokenizer.decode(self.token_cache, **self.decode_kwargs)
+            printable_text = text[self.print_len:]
+            self.token_cache = []
+            self.print_len = 0
+        else:
+            printable_text = ""
+        self.next_tokens_are_prompt = True
+        self.on_finalized_text(printable_text, stream_end=True)
+
+    # Disable base class's end hook; we'll finalize via manual_end()
+    def end(self):
+        pass
+```
+
+Example usage:
+
+```python
+streamer = BudgetAwareTextStreamer(
+    model.text_tokenizer,
+    skip_prompt=True,
+    skip_special_tokens=True
+)
+
+outputs = model.generate(
+    inputs=input_ids,
+    pixel_values=pixel_values,
+    grid_thws=grid_thws,
+    enable_thinking=enable_thinking,
+    enable_thinking_budget=enable_thinking_budget,
+    max_new_tokens=max_new_tokens,
+    thinking_budget=thinking_budget,
+    streamer=streamer
+)
+
+```
+
+</details>
+
+<details>
+<summary>Example: Multi-image</summary>
+Demonstrates how to run inference with multiple images and a related question.
+
+```python
+# Multi-image inference
+multi_image_files = [
+    "/path/to/image_1.jpg",
+    "/path/to/image_2.jpg",
+    "/path/to/image_3.jpg",
+]
+
+content = [{"type": "image", "image": Image.open(p).convert("RGB")} for p in multi_image_files]
+content.append({"type": "text", "text": "Describe the images."})
+messages = [{"role": "user", "content": content}]
+
+input_ids, pixel_values, grid_thws = model.preprocess_inputs(messages=messages, add_generation_prompt=True, max_pixels=896*896)
+input_ids = input_ids.cuda()
+pixel_values = pixel_values.cuda().to(model.dtype) if pixel_values is not None else None
+grid_thws = grid_thws.cuda() if grid_thws is not None else None
+
+with torch.no_grad():
+    outputs = model.generate(inputs=input_ids, pixel_values=pixel_values, grid_thws=grid_thws,
+                             max_new_tokens=1024, do_sample=True,
+                             eos_token_id=model.text_tokenizer.eos_token_id,
+                             pad_token_id=model.text_tokenizer.pad_token_id)
+print(model.text_tokenizer.decode(outputs[0], skip_special_tokens=True))
+```
+
+</details>
+
+<details>
+<summary>Example: Video</summary>
+Demonstrates how to run inference on a video by sampling multiple frames and asking the model to describe the content.
+
+```python
+# Video inference
+from moviepy.editor import VideoFileClip  # pip install moviepy==1.0.3
+
+video_file = "/path/to/video_1.mp4"
+num_frames = 8
+
+with VideoFileClip(video_file) as clip:
+    total_frames = int(clip.fps * clip.duration)
+    indices = [int(i * total_frames / num_frames) for i in range(num_frames)]
+    frames = [Image.fromarray(clip.get_frame(t)) for t in (idx / clip.fps for idx in indices)]
+
+messages = [{"role": "user", "content": [
+    {"type": "video", "video": frames},
+    {"type": "text", "text": "Describe this video in detail."},
+]}]
+
+input_ids, pixel_values, grid_thws = model.preprocess_inputs(messages=messages, add_generation_prompt=True, max_pixels=896*896)
+input_ids = input_ids.cuda()
+pixel_values = pixel_values.cuda().to(model.dtype) if pixel_values is not None else None
+grid_thws = grid_thws.cuda() if grid_thws is not None else None
+
+with torch.no_grad():
+    outputs = model.generate(inputs=input_ids, pixel_values=pixel_values, grid_thws=grid_thws,
+                             max_new_tokens=1024, do_sample=True,
+                             eos_token_id=model.text_tokenizer.eos_token_id,
+                             pad_token_id=model.text_tokenizer.pad_token_id)
+print(model.text_tokenizer.decode(outputs[0], skip_special_tokens=True))
+```
+
+</details>
+
+<details>
+<summary>Example: Text-only</summary>
+Demonstrates how to run inference using only text input without any images or videos.
+
+```python
+# Text-only inference
+messages = [{"role": "user", "content": "Hi, please introduce Yellow Mountain."}]
+
+input_ids, _, _ = model.preprocess_inputs(messages=messages, add_generation_prompt=True)
+input_ids = input_ids.cuda()
+
+with torch.no_grad():
+    outputs = model.generate(inputs=input_ids, max_new_tokens=1024, do_sample=True,
+                             eos_token_id=model.text_tokenizer.eos_token_id,
+                             pad_token_id=model.text_tokenizer.pad_token_id)
+print(model.text_tokenizer.decode(outputs[0], skip_special_tokens=True))
+```
+
+</details>
+
+
+To enable grounding, end your prompt with `Please provide the bounding box coordinates.` (for boxes) or `Please provide the point coordinates.` (for points). To target a specific object, wrap its description in `<ref>` tags, e.g.:
+
+```text
+Find the <ref>red apple</ref> in the image. Please provide the bounding box coordinates.
+```
+
+Coordinates are normalized to `[0,1)` with the origin `(0,0)` at the top-left corner of the image.
+
+* Point: `<point>(x,y)</point>`
+* Bounding box: `<box>(x1,y1),(x2,y2)</box>` where `(x1,y1)` is top-left, `(x2,y2)` is bottom-right.
+* Multiple results can be listed in square brackets: `[<box>(...)</box>,<box>(...)</box> ]`
+
+Example:
+
+```text
+The image features a serene scene with <ref>three birds</ref>[
+  <box>(0.401,0.526),(0.430,0.557)</box>,
+  <box>(0.489,0.494),(0.516,0.526)</box>,
+  <box>(0.296,0.529),(0.324,0.576)</box>
+] flying in formation against a clear blue sky.
+```
+
+
+
+## Model Zoo
+
+| Ovis MLLMs |           ViT           |          LLM          |                      Model Weights                      |                           Demo                           |
+|:-----------|:-----------------------:|:---------------------:|:-------------------------------------------------------:|:--------------------------------------------------------:|
+| Ovis2.5-2B   | siglip2-so400m-patch16-512 | Qwen3-1.7B | [Huggingface](https://huggingface.co/AIDC-AI/Ovis2.5-2B)  | [Space](https://huggingface.co/spaces/AIDC-AI/Ovis2.5-2B) |
+| Ovis2.5-9B   | siglip2-so400m-patch16-512  |  Qwen3-8B  | [Huggingface](https://huggingface.co/AIDC-AI/Ovis2.5-9B)  | [Space](https://huggingface.co/spaces/AIDC-AI/Ovis2.5-9B) |
+
+
+## Performance
+We evaluate Ovis2.5 using [VLMEvalKit](https://github.com/open-compass/VLMEvalKit), as employed in the OpenCompass multimodal and reasoning evaluation suite.
+
+![image/png](https://cdn-uploads.huggingface.co/production/uploads/637aebed7ce76c3b834cea37/LstPS8KqGObo03fCT5ezn.png)
+
+![image/png](https://cdn-uploads.huggingface.co/production/uploads/637aebed7ce76c3b834cea37/idTjwTPtGVO79x9I3iDPN.png)
+
+
+## Citation
+If you find Ovis useful, please consider citing the paper
+```bibtex
+@article{lu2025ovis25technicalreport,
+  title={Ovis2.5 Technical Report}, 
+  author={Shiyin Lu and Yang Li and Yu Xia and Yuwei Hu and Shanshan Zhao and Yanqing Ma and Zhichao Wei and Yinglun Li and Lunhao Duan and Jianshan Zhao and Yuxuan Han and Haijun Li and Wanying Chen and Junke Tang and Chengkun Hou and Zhixing Du and Tianli Zhou and Wenjie Zhang and Huping Ding and Jiahe Li and Wen Li and Gui Hu and Yiliang Gu and Siran Yang and Jiamang Wang and Hailong Sun and Yibo Wang and Hui Sun and Jinlong Huang and Yuping He and Shengze Shi and Weihong Zhang and Guodong Zheng and Junpeng Jiang and Sensen Gao and Yi-Feng Wu and Sijia Chen and Yuhui Chen and Qing-Guo Chen and Zhao Xu and Weihua Luo and Kaifu Zhang},
+  year={2025},
+  journal={arXiv:2508.11737}
+}
+
+@article{lu2024ovis,
+  title={Ovis: Structural Embedding Alignment for Multimodal Large Language Model},
+  author={Shiyin Lu and Yang Li and Qing-Guo Chen and Zhao Xu and Weihua Luo and Kaifu Zhang and Han-Jia Ye},
+  year={2024},
+  journal={arXiv:2405.20797}
+}
+```
+
+## License
+This project is licensed under the [Apache License, Version 2.0](https://www.apache.org/licenses/LICENSE-2.0.txt) (SPDX-License-Identifier: Apache-2.0).
+
+## Disclaimer
+We used compliance-checking algorithms during the training process, to ensure the compliance of the trained model to the best of our ability. Due to the complexity of the data and the diversity of language model usage scenarios, we cannot guarantee that the model is completely free of copyright issues or improper content. If you believe anything infringes on your rights or generates improper content, please contact us, and we will promptly address the matter.

added_tokens.json

@@ -0,0 +1,28 @@
+{
+  "</think>": 151668,
+  "</tool_call>": 151658,
+  "</tool_response>": 151666,
+  "<think>": 151667,
+  "<tool_call>": 151657,
+  "<tool_response>": 151665,
+  "<|box_end|>": 151649,
+  "<|box_start|>": 151648,
+  "<|endoftext|>": 151643,
+  "<|file_sep|>": 151664,
+  "<|fim_middle|>": 151660,
+  "<|fim_pad|>": 151662,
+  "<|fim_prefix|>": 151659,
+  "<|fim_suffix|>": 151661,
+  "<|im_end|>": 151645,
+  "<|im_start|>": 151644,
+  "<|image_pad|>": 151655,
+  "<|object_ref_end|>": 151647,
+  "<|object_ref_start|>": 151646,
+  "<|quad_end|>": 151651,
+  "<|quad_start|>": 151650,
+  "<|repo_name|>": 151663,
+  "<|video_pad|>": 151656,
+  "<|vision_end|>": 151653,
+  "<|vision_pad|>": 151654,
+  "<|vision_start|>": 151652
+}

chat_template.json

@@ -0,0 +1,3 @@
+{
+  "chat_template": "{%- for message in messages %}{{- '<|im_start|>' + message.role + '\n'}}{%- if message.role == 'system' or message.role == 'user' %}{%- if message.content is string %}{{- message.content | replace('<image>', '') | replace('<video>', '') }}{%- else %}{%- for item in message.content %}{%- if item.type == 'text' and 'text' in item %}{{- item.text | replace('<image>', '') | replace('<video>', '') }}{%- elif item.type == 'image' %}{{- '<image>'}}{%- elif item.type == 'video' %}{{- '<video>'}}{%- else %}{{- raise_exception('Invalid content type. Supported types for system and user are text, image, video.')}}{%- endif %}{%- if not loop.last %}{{- '\n'}}{%- endif %}{%- endfor %}{%- endif %}{%- elif message.role == 'assistant' %}{%- set content = '' %}{%- if message.content is string %}{%- set content = message.content | replace('<image>', '') | replace('<video>', '') %}{%- else %}{%- set ns = namespace(content='') -%}{%- for item in message.content %}{%- if item.type == 'text' and 'text' in item %}{%- set ns.content = ns.content ~ (item.text | replace('<image>', '') | replace('<video>', '')) %}{%- else %}{{- raise_exception('Invalid content type. Supported type for assistant is text.')}}{%- endif %}{%- endfor %}{%- set content = ns.content -%}{%- endif %}{%- set content = content.split('</think>')[-1].lstrip('\n') %}{{- content }}{%- else %}{{- raise_exception('Invalid role. Supported roles are system, user, assistant.')}}{%- endif %}{{- '<|im_end|>\n'}}{%- endfor %}{%- if add_generation_prompt %}{{- '<|im_start|>assistant\n' }}{%- if enable_thinking is defined and enable_thinking is false %}{{- '<think>\n\n</think>\n\n' }}{%- endif %}{%- endif %}"
+}

config.json

@@ -0,0 +1,72 @@
+{
+  "architectures": [
+      "Ovis2_5"
+  ],
+  "auto_map": {
+    "AutoConfig": "configuration_ovis2_5.Ovis2_5_Config",
+    "AutoModelForCausalLM": "modeling_ovis2_5.Ovis2_5"
+  },
+  "conversation_formatter_class": "Qwen3ConversationFormatter",
+  "hidden_size": 4096,
+  "vocab_size": 151936,
+  "num_attention_heads": 32,
+  "max_position_embeddings": 40960,
+  "llm_config": {
+      "_attn_implementation_autoset": true,
+      "_name_or_path": "Qwen/Qwen3-8B",
+      "architectures": [
+          "Qwen3ForCausalLM"
+      ],
+      "attention_bias": false,
+      "attention_dropout": 0.0,
+      "bos_token_id": 151643,
+      "eos_token_id": 151645,
+      "head_dim": 128,
+      "hidden_act": "silu",
+      "hidden_size": 4096,
+      "initializer_range": 0.02,
+      "intermediate_size": 12288,
+      "max_position_embeddings": 40960,
+      "max_window_layers": 36,
+      "model_type": "qwen3",
+      "num_attention_heads": 32,
+      "num_hidden_layers": 36,
+      "num_key_value_heads": 8,
+      "rms_norm_eps": 1e-06,
+      "rope_scaling": null,
+      "rope_theta": 1000000,
+      "sliding_window": null,
+      "torch_dtype": "float32",
+      "use_cache": true,
+      "use_sliding_window": false,
+      "vocab_size": 151936
+  },
+  "model_type": "ovis2_5",
+  "torch_dtype": "bfloat16",
+  "transformers_version": "4.51.3",
+  "use_cache": true,
+  "visual_vocab_size": 65536,
+  "vit_config": {
+      "_attn_implementation_autoset": true,
+      "_name_or_path": "google/siglip2-so400m-patch16-512", 
+      "attention_dropout": 0.0,
+      "fullatt_block_indexes": null,
+      "hidden_act": "gelu_pytorch_tanh",
+      "hidden_size": 1152,
+      "hidden_stride": 2,
+      "image_size": 512,
+      "intermediate_size": 4304,
+      "layer_norm_eps": 1e-06,
+      "model_type": "siglip2_navit",
+      "num_attention_heads": 16,
+      "num_channels": 3,
+      "num_hidden_layers": 27,
+      "num_patches": -1,
+      "patch_size": 16,
+      "preserve_original_pe": true,
+      "temporal_patch_size": 1,
+      "torch_dtype": "float32",
+      "use_rope": true,
+      "window_size": 112
+  }
+}

configuration.json

@@ -0,0 +1 @@
+{"framework": "pytorch", "task": "image-text-to-text", "allow_remote": true}

configuration_ovis2_5.py

@@ -0,0 +1,96 @@
+from typing import Any, Optional, List, Union
+
+from transformers import Qwen3Config
+from transformers.configuration_utils import PretrainedConfig
+
+__all__ = ["Siglip2NavitConfig", "Ovis2_5_Config"]
+
+
+class Siglip2NavitConfig(PretrainedConfig):
+    """This is the configuration class to store the configuration of an [`AIMv2Model`].
+
+    Instantiating a configuration with the defaults will yield a similar configuration
+    to that of the [apple/aimv2-large-patch14-224](https://huggingface.co/apple/aimv2-large-patch14-224).
+
+    Args:
+        hidden_size: Dimension of the hidden representations.
+        intermediate_size: Dimension of the SwiGLU representations.
+        num_hidden_layers: Number of hidden layers in the Transformer.
+        num_attention_heads: Number of attention heads for each attention layer
+            in the Transformer.
+        num_channels: Number of input channels.
+        image_size: Image size.
+        patch_size: Patch size.
+        rms_norm_eps: Epsilon value used for the RMS normalization layer.
+        attention_dropout: Dropout ratio for attention probabilities.
+        projection_dropout: Dropout ratio for the projection layer after the attention.
+        qkv_bias: Whether to add a bias to the queries, keys and values.
+        use_bias: Whether to add a bias in the feed-forward and projection layers.
+        kwargs: Keyword arguments for the [`PretrainedConfig`].
+    """
+
+    model_type: str = "siglip2_navit"
+
+    def __init__(
+        self,
+        hidden_size: int = 1024,
+        intermediate_size: int = 4096,
+        num_hidden_layers: int = 24,
+        num_attention_heads: int = 16,
+        num_channels: int = 3,
+        num_patches: int = -1,
+        image_size: int = 512,
+        patch_size: int = 16,
+        hidden_act: str="gelu_pytorch_tanh",
+        layer_norm_eps: float = 1e-6,
+        attention_dropout: float = 0.0,
+        hidden_stride: int = 2,
+        window_size: int = 112,
+        fullatt_block_indexes: Optional[list] = None,
+        temporal_patch_size: int = 1,
+        preserve_original_pe: bool = True,
+        use_rope: bool = True,
+        **kwargs: Any,
+    ):
+        super().__init__(**kwargs)
+        self.hidden_size = hidden_size
+        self.intermediate_size = intermediate_size
+        self.num_hidden_layers = num_hidden_layers
+        self.num_attention_heads = num_attention_heads
+        self.num_channels = num_channels
+        self.num_patches = num_patches
+        self.patch_size = patch_size
+        self.image_size = image_size
+        self.hidden_act = hidden_act
+        self.attention_dropout = attention_dropout
+        self.layer_norm_eps = layer_norm_eps
+        self.hidden_stride = hidden_stride
+        self.window_size = window_size
+        self.fullatt_block_indexes = fullatt_block_indexes
+        self.temporal_patch_size = temporal_patch_size
+        self.preserve_original_pe = preserve_original_pe
+        self.use_rope = use_rope
+
+class Ovis2_5_Config(PretrainedConfig):
+    model_type = "ovis2_5"
+    sub_configs = dict(llm_config=Qwen3Config, vit_config=Siglip2NavitConfig)
+
+    def __init__(self,
+        llm_config: Optional[Union[Qwen3Config, dict]] = None,
+        vit_config: Optional[Union[Siglip2NavitConfig, dict]] = None,
+        visual_vocab_size=65536,
+        hidden_size=None,
+        **kwargs
+    ):
+        super().__init__(**kwargs)
+        if isinstance(llm_config, dict):
+            llm_config = Qwen3Config(**llm_config)
+        self.llm_config = llm_config
+        if isinstance(vit_config, dict):
+            vit_config = Siglip2NavitConfig(**vit_config)
+        self.vit_config = vit_config
+        self.visual_vocab_size = visual_vocab_size
+        self.hidden_size = hidden_size
+        if kwargs.get('attn_implementation'):
+            self.llm_config._attn_implementation = kwargs['attn_implementation']
+            self.vit_config._attn_implementation = kwargs['attn_implementation']

generation_config.json

@@ -0,0 +1,15 @@
+{
+  "bos_token_id": 151643,
+  "do_sample": true,
+  "eos_token_id": [
+    151645,
+    151643
+  ],
+  "multimodal_max_length": 8192,
+  "pad_token_id": 151643,
+  "repetition_penalty": 1.05,
+  "temperature": 0.6,
+  "top_k": 20,
+  "top_p": 0.95,
+  "transformers_version": "4.51.3"
+}

model-00001-of-00004.safetensors

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

model-00002-of-00004.safetensors

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

model-00003-of-00004.safetensors

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

model-00004-of-00004.safetensors

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

model.safetensors.index.json

@@ -0,0 +1,847 @@
+{
+  "metadata": {
+    "total_size": 18349615568
+  },
+  "weight_map": {
+    "llm.lm_head.weight": "model-00001-of-00004.safetensors",
+    "llm.model.embed_tokens.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.0.input_layernorm.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.0.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.0.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.0.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.0.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.0.self_attn.k_norm.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.0.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.0.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.0.self_attn.q_norm.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.0.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.0.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.1.input_layernorm.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.1.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.1.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.1.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.1.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.1.self_attn.k_norm.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.1.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.1.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.1.self_attn.q_norm.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.1.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.1.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.10.input_layernorm.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.10.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.10.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.10.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.10.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.10.self_attn.k_norm.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.10.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.10.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.10.self_attn.q_norm.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.10.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.10.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.11.input_layernorm.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.11.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.11.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.11.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.11.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.11.self_attn.k_norm.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.11.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.11.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.11.self_attn.q_norm.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.11.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.11.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.12.input_layernorm.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.12.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.12.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.12.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.12.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.12.self_attn.k_norm.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.12.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.12.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.12.self_attn.q_norm.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.12.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.12.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.13.input_layernorm.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.13.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.13.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.13.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.13.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.13.self_attn.k_norm.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.13.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.13.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.13.self_attn.q_norm.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.13.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.13.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.14.input_layernorm.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.14.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
+    "llm.model.layers.14.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.14.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.14.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.14.self_attn.k_norm.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.14.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.14.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.14.self_attn.q_norm.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.14.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.14.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.15.input_layernorm.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.15.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.15.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.15.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.15.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.15.self_attn.k_norm.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.15.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.15.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.15.self_attn.q_norm.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.15.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.15.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.16.input_layernorm.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.16.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.16.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.16.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.16.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.16.self_attn.k_norm.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.16.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.16.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.16.self_attn.q_norm.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.16.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.16.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.17.input_layernorm.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.17.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.17.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.17.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.17.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.17.self_attn.k_norm.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.17.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.17.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.17.self_attn.q_norm.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.17.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.17.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.18.input_layernorm.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.18.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.18.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.18.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.18.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.18.self_attn.k_norm.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.18.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.18.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.18.self_attn.q_norm.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.18.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.18.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.19.input_layernorm.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.19.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.19.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.19.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.19.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.19.self_attn.k_norm.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.19.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.19.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.19.self_attn.q_norm.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.19.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.19.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.2.input_layernorm.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.2.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.2.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.2.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.2.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.2.self_attn.k_norm.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.2.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.2.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.2.self_attn.q_norm.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.2.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.2.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.20.input_layernorm.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.20.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.20.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.20.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.20.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.20.self_attn.k_norm.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.20.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.20.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.20.self_attn.q_norm.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.20.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.20.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.21.input_layernorm.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.21.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.21.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.21.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.21.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.21.self_attn.k_norm.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.21.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.21.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.21.self_attn.q_norm.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.21.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.21.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.22.input_layernorm.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.22.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.22.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.22.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.22.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.22.self_attn.k_norm.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.22.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.22.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.22.self_attn.q_norm.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.22.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.22.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.23.input_layernorm.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.23.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.23.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.23.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.23.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.23.self_attn.k_norm.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.23.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.23.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.23.self_attn.q_norm.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.23.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.23.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.24.input_layernorm.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.24.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.24.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.24.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.24.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.24.self_attn.k_norm.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.24.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.24.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.24.self_attn.q_norm.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.24.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.24.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.25.input_layernorm.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.25.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.25.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.25.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.25.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.25.self_attn.k_norm.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.25.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.25.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.25.self_attn.q_norm.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.25.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.25.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.26.input_layernorm.weight": "model-00002-of-00004.safetensors",
+    "llm.model.layers.26.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.26.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.26.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.26.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.26.self_attn.k_norm.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.26.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.26.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.26.self_attn.q_norm.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.26.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.26.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.27.input_layernorm.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.27.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.27.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.27.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.27.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.27.self_attn.k_norm.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.27.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.27.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.27.self_attn.q_norm.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.27.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.27.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.28.input_layernorm.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.28.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.28.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.28.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.28.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.28.self_attn.k_norm.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.28.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.28.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.28.self_attn.q_norm.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.28.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.28.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.29.input_layernorm.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.29.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.29.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.29.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.29.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.29.self_attn.k_norm.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.29.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.29.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.29.self_attn.q_norm.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.29.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.29.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.3.input_layernorm.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.3.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.3.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.3.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.3.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.3.self_attn.k_norm.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.3.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.3.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.3.self_attn.q_norm.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.3.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.3.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.30.input_layernorm.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.30.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.30.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.30.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.30.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.30.self_attn.k_norm.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.30.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.30.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.30.self_attn.q_norm.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.30.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.30.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.31.input_layernorm.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.31.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.31.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.31.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.31.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.31.self_attn.k_norm.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.31.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.31.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.31.self_attn.q_norm.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.31.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.31.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.32.input_layernorm.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.32.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.32.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.32.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.32.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.32.self_attn.k_norm.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.32.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.32.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.32.self_attn.q_norm.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.32.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.32.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.33.input_layernorm.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.33.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.33.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.33.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.33.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.33.self_attn.k_norm.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.33.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.33.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.33.self_attn.q_norm.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.33.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.33.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.34.input_layernorm.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.34.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.34.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.34.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.34.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.34.self_attn.k_norm.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.34.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.34.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.34.self_attn.q_norm.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.34.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.34.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.35.input_layernorm.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.35.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.35.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.35.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.35.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.35.self_attn.k_norm.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.35.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.35.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.35.self_attn.q_norm.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.35.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.35.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.4.input_layernorm.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.4.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.4.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.4.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.4.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.4.self_attn.k_norm.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.4.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.4.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.4.self_attn.q_norm.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.4.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.4.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.5.input_layernorm.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.5.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.5.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.5.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.5.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.5.self_attn.k_norm.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.5.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.5.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.5.self_attn.q_norm.weight": "model-00003-of-00004.safetensors",
+    "llm.model.layers.5.self_attn.q_proj.weight": "model-00004-of-00004.safetensors",
+    "llm.model.layers.5.self_attn.v_proj.weight": "model-00004-of-00004.safetensors",
+    "llm.model.layers.6.input_layernorm.weight": "model-00004-of-00004.safetensors",
+    "llm.model.layers.6.mlp.down_proj.weight": "model-00004-of-00004.safetensors",
+    "llm.model.layers.6.mlp.gate_proj.weight": "model-00004-of-00004.safetensors",
+    "llm.model.layers.6.mlp.up_proj.weight": "model-00004-of-00004.safetensors",
+    "llm.model.layers.6.post_attention_layernorm.weight": "model-00004-of-00004.safetensors",
+    "llm.model.layers.6.self_attn.k_norm.weight": "model-00004-of-00004.safetensors",
+    "llm.model.layers.6.self_attn.k_proj.weight": "model-00004-of-00004.safetensors",
+    "llm.model.layers.6.self_attn.o_proj.weight": "model-00004-of-00004.safetensors",
+    "llm.model.layers.6.self_attn.q_norm.weight": "model-00004-of-00004.safetensors",
+    "llm.model.layers.6.self_attn.q_proj.weight": "model-00004-of-00004.safetensors",
+    "llm.model.layers.6.self_attn.v_proj.weight": "model-00004-of-00004.safetensors",
+    "llm.model.layers.7.input_layernorm.weight": "model-00004-of-00004.safetensors",
+    "llm.model.layers.7.mlp.down_proj.weight": "model-00004-of-00004.safetensors",
+    "llm.model.layers.7.mlp.gate_proj.weight": "model-00004-of-00004.safetensors",
+    "llm.model.layers.7.mlp.up_proj.weight": "model-00004-of-00004.safetensors",
+    "llm.model.layers.7.post_attention_layernorm.weight": "model-00004-of-00004.safetensors",
+    "llm.model.layers.7.self_attn.k_norm.weight": "model-00004-of-00004.safetensors",
+    "llm.model.layers.7.self_attn.k_proj.weight": "model-00004-of-00004.safetensors",
+    "llm.model.layers.7.self_attn.o_proj.weight": "model-00004-of-00004.safetensors",
+    "llm.model.layers.7.self_attn.q_norm.weight": "model-00004-of-00004.safetensors",
+    "llm.model.layers.7.self_attn.q_proj.weight": "model-00004-of-00004.safetensors",
+    "llm.model.layers.7.self_attn.v_proj.weight": "model-00004-of-00004.safetensors",
+    "llm.model.layers.8.input_layernorm.weight": "model-00004-of-00004.safetensors",
+    "llm.model.layers.8.mlp.down_proj.weight": "model-00004-of-00004.safetensors",
+    "llm.model.layers.8.mlp.gate_proj.weight": "model-00004-of-00004.safetensors",
+    "llm.model.layers.8.mlp.up_proj.weight": "model-00004-of-00004.safetensors",
+    "llm.model.layers.8.post_attention_layernorm.weight": "model-00004-of-00004.safetensors",
+    "llm.model.layers.8.self_attn.k_norm.weight": "model-00004-of-00004.safetensors",
+    "llm.model.layers.8.self_attn.k_proj.weight": "model-00004-of-00004.safetensors",
+    "llm.model.layers.8.self_attn.o_proj.weight": "model-00004-of-00004.safetensors",
+    "llm.model.layers.8.self_attn.q_norm.weight": "model-00004-of-00004.safetensors",
+    "llm.model.layers.8.self_attn.q_proj.weight": "model-00004-of-00004.safetensors",
+    "llm.model.layers.8.self_attn.v_proj.weight": "model-00004-of-00004.safetensors",
+    "llm.model.layers.9.input_layernorm.weight": "model-00004-of-00004.safetensors",
+    "llm.model.layers.9.mlp.down_proj.weight": "model-00004-of-00004.safetensors",
+    "llm.model.layers.9.mlp.gate_proj.weight": "model-00004-of-00004.safetensors",
+    "llm.model.layers.9.mlp.up_proj.weight": "model-00004-of-00004.safetensors",
+    "llm.model.layers.9.post_attention_layernorm.weight": "model-00004-of-00004.safetensors",
+    "llm.model.layers.9.self_attn.k_norm.weight": "model-00004-of-00004.safetensors",
+    "llm.model.layers.9.self_attn.k_proj.weight": "model-00004-of-00004.safetensors",
+    "llm.model.layers.9.self_attn.o_proj.weight": "model-00004-of-00004.safetensors",
+    "llm.model.layers.9.self_attn.q_norm.weight": "model-00004-of-00004.safetensors",
+    "llm.model.layers.9.self_attn.q_proj.weight": "model-00004-of-00004.safetensors",
+    "llm.model.layers.9.self_attn.v_proj.weight": "model-00004-of-00004.safetensors",
+    "llm.model.norm.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.head.0.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.head.1.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.head.1.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.embeddings.patch_embedding.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.embeddings.patch_embedding.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.embeddings.position_embedding.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.0.layer_norm1.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.0.layer_norm1.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.0.layer_norm2.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.0.layer_norm2.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.0.mlp.fc1.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.0.mlp.fc1.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.0.mlp.fc2.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.0.mlp.fc2.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.0.self_attn.k_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.0.self_attn.k_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.0.self_attn.out_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.0.self_attn.out_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.0.self_attn.q_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.0.self_attn.q_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.0.self_attn.v_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.0.self_attn.v_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.1.layer_norm1.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.1.layer_norm1.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.1.layer_norm2.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.1.layer_norm2.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.1.mlp.fc1.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.1.mlp.fc1.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.1.mlp.fc2.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.1.mlp.fc2.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.1.self_attn.k_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.1.self_attn.k_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.1.self_attn.out_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.1.self_attn.out_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.1.self_attn.q_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.1.self_attn.q_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.1.self_attn.v_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.1.self_attn.v_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.10.layer_norm1.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.10.layer_norm1.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.10.layer_norm2.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.10.layer_norm2.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.10.mlp.fc1.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.10.mlp.fc1.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.10.mlp.fc2.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.10.mlp.fc2.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.10.self_attn.k_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.10.self_attn.k_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.10.self_attn.out_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.10.self_attn.out_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.10.self_attn.q_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.10.self_attn.q_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.10.self_attn.v_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.10.self_attn.v_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.11.layer_norm1.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.11.layer_norm1.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.11.layer_norm2.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.11.layer_norm2.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.11.mlp.fc1.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.11.mlp.fc1.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.11.mlp.fc2.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.11.mlp.fc2.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.11.self_attn.k_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.11.self_attn.k_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.11.self_attn.out_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.11.self_attn.out_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.11.self_attn.q_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.11.self_attn.q_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.11.self_attn.v_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.11.self_attn.v_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.12.layer_norm1.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.12.layer_norm1.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.12.layer_norm2.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.12.layer_norm2.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.12.mlp.fc1.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.12.mlp.fc1.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.12.mlp.fc2.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.12.mlp.fc2.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.12.self_attn.k_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.12.self_attn.k_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.12.self_attn.out_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.12.self_attn.out_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.12.self_attn.q_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.12.self_attn.q_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.12.self_attn.v_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.12.self_attn.v_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.13.layer_norm1.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.13.layer_norm1.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.13.layer_norm2.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.13.layer_norm2.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.13.mlp.fc1.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.13.mlp.fc1.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.13.mlp.fc2.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.13.mlp.fc2.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.13.self_attn.k_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.13.self_attn.k_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.13.self_attn.out_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.13.self_attn.out_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.13.self_attn.q_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.13.self_attn.q_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.13.self_attn.v_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.13.self_attn.v_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.14.layer_norm1.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.14.layer_norm1.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.14.layer_norm2.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.14.layer_norm2.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.14.mlp.fc1.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.14.mlp.fc1.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.14.mlp.fc2.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.14.mlp.fc2.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.14.self_attn.k_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.14.self_attn.k_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.14.self_attn.out_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.14.self_attn.out_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.14.self_attn.q_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.14.self_attn.q_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.14.self_attn.v_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.14.self_attn.v_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.15.layer_norm1.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.15.layer_norm1.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.15.layer_norm2.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.15.layer_norm2.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.15.mlp.fc1.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.15.mlp.fc1.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.15.mlp.fc2.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.15.mlp.fc2.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.15.self_attn.k_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.15.self_attn.k_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.15.self_attn.out_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.15.self_attn.out_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.15.self_attn.q_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.15.self_attn.q_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.15.self_attn.v_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.15.self_attn.v_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.16.layer_norm1.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.16.layer_norm1.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.16.layer_norm2.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.16.layer_norm2.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.16.mlp.fc1.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.16.mlp.fc1.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.16.mlp.fc2.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.16.mlp.fc2.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.16.self_attn.k_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.16.self_attn.k_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.16.self_attn.out_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.16.self_attn.out_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.16.self_attn.q_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.16.self_attn.q_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.16.self_attn.v_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.16.self_attn.v_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.17.layer_norm1.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.17.layer_norm1.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.17.layer_norm2.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.17.layer_norm2.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.17.mlp.fc1.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.17.mlp.fc1.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.17.mlp.fc2.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.17.mlp.fc2.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.17.self_attn.k_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.17.self_attn.k_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.17.self_attn.out_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.17.self_attn.out_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.17.self_attn.q_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.17.self_attn.q_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.17.self_attn.v_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.17.self_attn.v_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.18.layer_norm1.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.18.layer_norm1.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.18.layer_norm2.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.18.layer_norm2.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.18.mlp.fc1.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.18.mlp.fc1.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.18.mlp.fc2.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.18.mlp.fc2.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.18.self_attn.k_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.18.self_attn.k_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.18.self_attn.out_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.18.self_attn.out_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.18.self_attn.q_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.18.self_attn.q_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.18.self_attn.v_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.18.self_attn.v_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.19.layer_norm1.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.19.layer_norm1.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.19.layer_norm2.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.19.layer_norm2.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.19.mlp.fc1.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.19.mlp.fc1.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.19.mlp.fc2.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.19.mlp.fc2.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.19.self_attn.k_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.19.self_attn.k_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.19.self_attn.out_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.19.self_attn.out_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.19.self_attn.q_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.19.self_attn.q_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.19.self_attn.v_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.19.self_attn.v_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.2.layer_norm1.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.2.layer_norm1.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.2.layer_norm2.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.2.layer_norm2.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.2.mlp.fc1.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.2.mlp.fc1.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.2.mlp.fc2.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.2.mlp.fc2.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.2.self_attn.k_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.2.self_attn.k_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.2.self_attn.out_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.2.self_attn.out_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.2.self_attn.q_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.2.self_attn.q_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.2.self_attn.v_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.2.self_attn.v_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.20.layer_norm1.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.20.layer_norm1.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.20.layer_norm2.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.20.layer_norm2.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.20.mlp.fc1.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.20.mlp.fc1.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.20.mlp.fc2.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.20.mlp.fc2.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.20.self_attn.k_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.20.self_attn.k_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.20.self_attn.out_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.20.self_attn.out_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.20.self_attn.q_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.20.self_attn.q_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.20.self_attn.v_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.20.self_attn.v_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.21.layer_norm1.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.21.layer_norm1.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.21.layer_norm2.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.21.layer_norm2.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.21.mlp.fc1.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.21.mlp.fc1.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.21.mlp.fc2.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.21.mlp.fc2.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.21.self_attn.k_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.21.self_attn.k_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.21.self_attn.out_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.21.self_attn.out_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.21.self_attn.q_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.21.self_attn.q_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.21.self_attn.v_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.21.self_attn.v_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.22.layer_norm1.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.22.layer_norm1.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.22.layer_norm2.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.22.layer_norm2.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.22.mlp.fc1.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.22.mlp.fc1.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.22.mlp.fc2.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.22.mlp.fc2.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.22.self_attn.k_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.22.self_attn.k_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.22.self_attn.out_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.22.self_attn.out_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.22.self_attn.q_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.22.self_attn.q_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.22.self_attn.v_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.22.self_attn.v_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.23.layer_norm1.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.23.layer_norm1.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.23.layer_norm2.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.23.layer_norm2.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.23.mlp.fc1.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.23.mlp.fc1.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.23.mlp.fc2.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.23.mlp.fc2.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.23.self_attn.k_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.23.self_attn.k_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.23.self_attn.out_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.23.self_attn.out_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.23.self_attn.q_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.23.self_attn.q_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.23.self_attn.v_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.23.self_attn.v_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.24.layer_norm1.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.24.layer_norm1.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.24.layer_norm2.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.24.layer_norm2.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.24.mlp.fc1.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.24.mlp.fc1.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.24.mlp.fc2.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.24.mlp.fc2.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.24.self_attn.k_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.24.self_attn.k_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.24.self_attn.out_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.24.self_attn.out_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.24.self_attn.q_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.24.self_attn.q_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.24.self_attn.v_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.24.self_attn.v_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.25.layer_norm1.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.25.layer_norm1.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.25.layer_norm2.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.25.layer_norm2.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.25.mlp.fc1.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.25.mlp.fc1.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.25.mlp.fc2.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.25.mlp.fc2.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.25.self_attn.k_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.25.self_attn.k_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.25.self_attn.out_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.25.self_attn.out_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.25.self_attn.q_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.25.self_attn.q_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.25.self_attn.v_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.25.self_attn.v_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.26.layer_norm1.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.26.layer_norm1.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.26.layer_norm2.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.26.layer_norm2.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.26.mlp.fc1.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.26.mlp.fc1.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.26.mlp.fc2.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.26.mlp.fc2.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.26.self_attn.k_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.26.self_attn.k_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.26.self_attn.out_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.26.self_attn.out_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.26.self_attn.q_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.26.self_attn.q_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.26.self_attn.v_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.26.self_attn.v_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.3.layer_norm1.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.3.layer_norm1.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.3.layer_norm2.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.3.layer_norm2.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.3.mlp.fc1.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.3.mlp.fc1.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.3.mlp.fc2.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.3.mlp.fc2.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.3.self_attn.k_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.3.self_attn.k_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.3.self_attn.out_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.3.self_attn.out_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.3.self_attn.q_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.3.self_attn.q_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.3.self_attn.v_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.3.self_attn.v_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.4.layer_norm1.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.4.layer_norm1.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.4.layer_norm2.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.4.layer_norm2.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.4.mlp.fc1.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.4.mlp.fc1.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.4.mlp.fc2.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.4.mlp.fc2.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.4.self_attn.k_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.4.self_attn.k_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.4.self_attn.out_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.4.self_attn.out_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.4.self_attn.q_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.4.self_attn.q_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.4.self_attn.v_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.4.self_attn.v_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.5.layer_norm1.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.5.layer_norm1.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.5.layer_norm2.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.5.layer_norm2.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.5.mlp.fc1.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.5.mlp.fc1.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.5.mlp.fc2.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.5.mlp.fc2.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.5.self_attn.k_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.5.self_attn.k_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.5.self_attn.out_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.5.self_attn.out_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.5.self_attn.q_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.5.self_attn.q_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.5.self_attn.v_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.5.self_attn.v_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.6.layer_norm1.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.6.layer_norm1.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.6.layer_norm2.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.6.layer_norm2.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.6.mlp.fc1.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.6.mlp.fc1.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.6.mlp.fc2.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.6.mlp.fc2.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.6.self_attn.k_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.6.self_attn.k_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.6.self_attn.out_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.6.self_attn.out_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.6.self_attn.q_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.6.self_attn.q_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.6.self_attn.v_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.6.self_attn.v_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.7.layer_norm1.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.7.layer_norm1.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.7.layer_norm2.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.7.layer_norm2.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.7.mlp.fc1.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.7.mlp.fc1.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.7.mlp.fc2.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.7.mlp.fc2.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.7.self_attn.k_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.7.self_attn.k_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.7.self_attn.out_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.7.self_attn.out_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.7.self_attn.q_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.7.self_attn.q_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.7.self_attn.v_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.7.self_attn.v_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.8.layer_norm1.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.8.layer_norm1.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.8.layer_norm2.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.8.layer_norm2.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.8.mlp.fc1.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.8.mlp.fc1.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.8.mlp.fc2.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.8.mlp.fc2.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.8.self_attn.k_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.8.self_attn.k_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.8.self_attn.out_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.8.self_attn.out_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.8.self_attn.q_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.8.self_attn.q_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.8.self_attn.v_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.8.self_attn.v_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.9.layer_norm1.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.9.layer_norm1.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.9.layer_norm2.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.9.layer_norm2.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.9.mlp.fc1.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.9.mlp.fc1.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.9.mlp.fc2.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.9.mlp.fc2.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.9.self_attn.k_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.9.self_attn.k_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.9.self_attn.out_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.9.self_attn.out_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.9.self_attn.q_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.9.self_attn.q_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.9.self_attn.v_proj.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.encoder.layers.9.self_attn.v_proj.weight": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.post_layernorm.bias": "model-00004-of-00004.safetensors",
+    "visual_tokenizer.vit.vision_model.post_layernorm.weight": "model-00004-of-00004.safetensors",
+    "vte.weight": "model-00004-of-00004.safetensors"
+  }
+}

modeling_ovis2_5.py

@@ -0,0 +1,999 @@
+import math
+from typing import Dict, List, Optional, Tuple, Union
+
+import PIL.Image
+import numpy as np
+import torch
+from torch import Tensor, nn
+from torch.nn import functional as F
+from transformers import (
+    AutoConfig,
+    AutoImageProcessor,
+    AutoModel,
+    AutoModelForCausalLM,
+    AutoTokenizer,
+)
+from transformers.activations import ACT2FN
+from transformers.generation.utils import GenerateOutput
+from transformers.modeling_outputs import BaseModelOutputWithNoAttention
+from transformers.modeling_utils import PreTrainedModel
+from transformers.utils import is_flash_attn_2_available
+
+from .configuration_ovis2_5 import Siglip2NavitConfig, Ovis2_5_Config
+
+
+if is_flash_attn_2_available():
+    from flash_attn import flash_attn_varlen_func
+    from flash_attn.layers.rotary import apply_rotary_emb
+
+
+IMAGE_PLACEHOLDER = "<image>"
+IMAGE_PLACEHOLDER_ID = -200
+VIDEO_PLACEHOLDER = "<video>"
+VIDEO_PLACEHOLDER_ID = -201
+
+VISUAL_ATOM_ID = -300
+INDICATOR_IDS = [-301, -302, -303, -304]
+
+
+# copied from qwen2.5-vl
+class VisionRotaryEmbedding(nn.Module):
+    def __init__(self, dim: int, theta: float = 10000.0) -> None:
+        super().__init__()
+        inv_freq = 1.0 / (theta ** (torch.arange(0, dim, 2, dtype=torch.float) / dim))
+        self.register_buffer("inv_freq", inv_freq, persistent=False)
+
+    def forward(self, seqlen: int) -> torch.Tensor:
+        seq = torch.arange(seqlen, device=self.inv_freq.device, dtype=self.inv_freq.dtype)
+        freqs = torch.outer(seq, self.inv_freq)
+        return freqs
+
+
+class Siglip2VisionEmbeddings(nn.Module):
+    def __init__(self, config: Siglip2NavitConfig):
+        super().__init__()
+        self.config = config
+        self.embed_dim = config.hidden_size
+        self.patch_size = config.patch_size
+        self.image_size = config.image_size
+        self.num_patches = config.num_patches
+        self.preserve_original_pe = config.preserve_original_pe
+        self.hidden_stride = config.hidden_stride
+
+
+        # siglip2 naflex
+        if self.num_patches > 0:
+            self.patch_embedding = nn.Linear(
+                    in_features=config.num_channels * self.patch_size * self.patch_size,
+                    out_features=self.embed_dim,
+                )
+            if self.preserve_original_pe:
+                self.position_embedding_size = int(self.num_patches**0.5)
+                self.position_embedding = nn.Embedding(self.num_patches, self.embed_dim)
+
+        else:
+            self.patch_embedding = nn.Conv2d(
+                    in_channels=config.num_channels,
+                    out_channels=self.embed_dim,
+                    kernel_size=self.patch_size,
+                    stride=self.patch_size,
+                    padding="valid",
+                )
+            if self.preserve_original_pe:
+                self.num_patches = (self.image_size // self.patch_size) ** 2
+                self.position_embedding_size = self.image_size // self.patch_size
+                self.position_embedding = nn.Embedding(self.num_patches, self.embed_dim)
+        
+    @staticmethod
+    def resize_positional_embeddings(
+        positional_embeddings: torch.Tensor,
+        spatial_shapes: torch.LongTensor,
+        max_length: int,
+    ) -> torch.Tensor:
+        """
+        Resize positional embeddings to image-specific size and pad to a fixed size.
+        Args:
+            positional_embeddings (`torch.Tensor`):
+                Position embeddings of shape (height, width, embed_dim)
+            spatial_shapes (`torch.LongTensor`):
+                Spatial shapes of shape (batch_size, 2) to resize the positional embeddings to
+            max_length (`int`):
+                Maximum length of the positional embeddings to pad resized positional embeddings to
+        Returns:
+            `torch.Tensor`: Embeddings of shape (batch_size, max_length, embed_dim)
+        """
+        batch_size = spatial_shapes.shape[0]
+        embed_dim = positional_embeddings.shape[-1]
+        source_dtype = positional_embeddings.dtype
+
+        resulted_positional_embeddings = torch.empty(
+            (batch_size, max_length, embed_dim),
+            device=positional_embeddings.device,
+            dtype=source_dtype,
+        )
+
+        # (height, width, embed_dim) -> (1, embed_dim, height, width) for interpolation
+        positional_embeddings = positional_embeddings.permute(2, 0, 1).unsqueeze(0)
+
+        # Upcast to float32 on CPU because antialias is not supported for bfloat16/float16 on CPU
+        if positional_embeddings.device.type == "cpu":
+            positional_embeddings = positional_embeddings.to(torch.float32)
+
+        for i in range(batch_size):
+            # (1, dim, height, width) -> (1, dim, target_height, target_width)
+            height, width = spatial_shapes[i]
+            resized_embeddings = F.interpolate(
+                positional_embeddings,
+                size=(height, width),
+                mode="bilinear",
+                align_corners=False,
+                antialias=True,
+            )
+
+            # (1, dim, target_height, target_width) -> (target_height * target_width, dim)
+            resized_embeddings = resized_embeddings.reshape(embed_dim, height * width).transpose(0, 1)
+
+            # Cast to original dtype
+            resized_embeddings = resized_embeddings.to(source_dtype)
+
+            resulted_positional_embeddings[i, : height * width] = resized_embeddings
+            resulted_positional_embeddings[i, height * width :] = resized_embeddings[0]
+
+        return resulted_positional_embeddings
+
+    def forward(self, pixel_values: torch.FloatTensor, 
+                grid_thws: Optional[torch.LongTensor] = None) -> torch.Tensor:
+        """
+        Args:
+            pixel_values (`torch.FloatTensor`):
+                Pixel values of shape (num_patches, num_channels * temporal_patch_size * patch_size * patch_size)
+            grid_thws: (`torch.LongTensor`):
+                grid shape (num_patches, 3)
+        """
+
+        # Apply patch embeddings to already patchified pixel values
+        target_dtype = self.patch_embedding.weight.dtype
+        if isinstance(self.patch_embedding, nn.Linear):
+            patch_embeds = self.patch_embedding(pixel_values.to(dtype=target_dtype))
+        elif isinstance(self.patch_embedding, nn.Conv2d):
+            pixel_values = pixel_values.view(-1, self.config.num_channels * self.config.temporal_patch_size, self.patch_size,
+                   self.patch_size)
+            patch_embeds = self.patch_embedding(pixel_values.to(dtype=target_dtype))
+            patch_embeds = patch_embeds.reshape(-1, self.embed_dim)
+
+
+        if self.preserve_original_pe:
+            assert grid_thws is not None
+            pos_embed_new = torch.zeros_like(patch_embeds)
+            ori_h = ori_w = self.position_embedding_size
+            positional_embeddings = self.position_embedding.weight.reshape(
+                                self.position_embedding_size, self.position_embedding_size, -1
+                            ).unsqueeze(0).permute(0,3,1,2)
+            # pos_embed = self.pos_embed.reshape(1, ori_h, ori_w, -1).permute(0, 3, 1, 2)
+            cnt = 0
+            for t, h, w in grid_thws:
+                thw = t * h * w
+                pe = F.interpolate(positional_embeddings, size=(h, w), mode='bicubic', align_corners=False)
+                pe = pe.permute(0, 2, 3, 1).reshape(1, h * w, -1)
+                pe = pe[0].repeat(t, 1)
+                pe = pe.reshape(t, h // self.hidden_stride, self.hidden_stride, w // self.hidden_stride,
+                                self.hidden_stride, -1)
+                pe = pe.permute(0, 1, 3, 2, 4, 5).reshape(thw, -1)
+                pos_embed_new[cnt:cnt + thw] = pe
+                cnt += thw
+            patch_embeds = patch_embeds + pos_embed_new
+
+        return patch_embeds
+
+
+# copied from qwen2.5-vl
+def apply_rotary_pos_emb_flashatt(
+    q: torch.Tensor, k: torch.Tensor, cos: torch.Tensor, sin: torch.Tensor
+) -> Tuple[torch.Tensor, torch.Tensor]:
+    cos = cos.chunk(2, dim=-1)[0].contiguous()
+    sin = sin.chunk(2, dim=-1)[0].contiguous()
+    q_embed = apply_rotary_emb(q.float(), cos.float(), sin.float()).type_as(q)
+    k_embed = apply_rotary_emb(k.float(), cos.float(), sin.float()).type_as(k)
+    return q_embed, k_embed
+
+
+# Copied from transformers.models.llama.modeling_llama.rotate_half
+def rotate_half(x):
+    """Rotates half the hidden dims of the input."""
+    x1 = x[..., : x.shape[-1] // 2]
+    x2 = x[..., x.shape[-1] // 2 :]
+    return torch.cat((-x2, x1), dim=-1)
+
+
+def apply_rotary_pos_emb_vision(
+    q: torch.Tensor, k: torch.Tensor, cos: torch.Tensor, sin: torch.Tensor
+) -> tuple[torch.Tensor, torch.Tensor]:
+    orig_q_dtype = q.dtype
+    orig_k_dtype = k.dtype
+    q, k = q.float(), k.float()
+    cos, sin = cos.unsqueeze(-2).float(), sin.unsqueeze(-2).float()
+    q_embed = (q * cos) + (rotate_half(q) * sin)
+    k_embed = (k * cos) + (rotate_half(k) * sin)
+    q_embed = q_embed.to(orig_q_dtype)
+    k_embed = k_embed.to(orig_k_dtype)
+    return q_embed, k_embed
+
+
+class Siglip2Attention(nn.Module):
+    """Multi-headed attention from 'Attention Is All You Need' paper"""
+
+    def __init__(self, config):
+        super().__init__()
+        self.config = config
+        self.embed_dim = config.hidden_size
+        self.num_heads = config.num_attention_heads
+        self.head_dim = self.embed_dim // self.num_heads
+        if self.head_dim * self.num_heads != self.embed_dim:
+            raise ValueError(
+                f"embed_dim must be divisible by num_heads (got `embed_dim`: {self.embed_dim} and `num_heads`:"
+                f" {self.num_heads})."
+            )
+        self.scale = self.head_dim**-0.5
+        self.dropout = config.attention_dropout
+        self.is_causal = False
+
+        self.k_proj = nn.Linear(self.embed_dim, self.embed_dim)
+        self.v_proj = nn.Linear(self.embed_dim, self.embed_dim)
+        self.q_proj = nn.Linear(self.embed_dim, self.embed_dim)
+        self.out_proj = nn.Linear(self.embed_dim, self.embed_dim)
+
+        self.use_rope = config.use_rope
+
+    def forward(
+        self,
+        hidden_states: torch.Tensor,
+        cu_seqlens: torch.Tensor,
+        position_embeddings: Optional[Tuple[torch.Tensor, torch.Tensor]] = None,
+    ) -> tuple[torch.Tensor, Optional[torch.Tensor]]:
+        """Input shape: Batch x Time x Channel"""
+
+        seq_length, embed_dim = hidden_states.shape
+
+        queries = self.q_proj(hidden_states)
+        keys = self.k_proj(hidden_states)
+        values = self.v_proj(hidden_states)
+
+        queries = queries.view(seq_length, self.num_heads, self.head_dim)
+        keys = keys.view(seq_length, self.num_heads, self.head_dim)
+        values = values.view(seq_length, self.num_heads, self.head_dim)
+
+        if self.use_rope:
+            cos, sin = position_embeddings
+            if is_flash_attn_2_available():
+                queries, keys = apply_rotary_pos_emb_flashatt(queries.unsqueeze(0), keys.unsqueeze(0), cos, sin)
+            else:
+                queries, keys = apply_rotary_pos_emb_vision(queries.unsqueeze(0), keys.unsqueeze(0), cos, sin)
+            queries = queries.squeeze(0)
+            keys = keys.squeeze(0)
+
+        max_seqlen = (cu_seqlens[1:] - cu_seqlens[:-1]).max().item()
+        if is_flash_attn_2_available():
+            attn_output = flash_attn_varlen_func(queries, keys, values, cu_seqlens, cu_seqlens, max_seqlen, max_seqlen).reshape(
+                                                seq_length, -1
+                                            )
+        else:
+            batch_size = cu_seqlens.shape[0] - 1
+            outputs = []
+            cu = cu_seqlens.tolist()
+            for i in range(batch_size):
+                start_idx = cu[i]
+                end_idx = cu[i + 1]
+                # Each sequence is processed independently.
+                q_i = queries[start_idx:end_idx].unsqueeze(0)
+                k_i = keys[start_idx:end_idx].unsqueeze(0)
+                v_i = values[start_idx:end_idx].unsqueeze(0)
+                # (1, seq_len, num_heads, head_dim) ->
+                # (1, num_heads, seq_len, head_dim)
+                q_i, k_i, v_i = [x.transpose(1, 2) for x in (q_i, k_i, v_i)]
+                output_i = F.scaled_dot_product_attention(q_i,
+                                                        k_i,
+                                                        v_i,
+                                                        dropout_p=0.0)
+                # (1, num_heads, seq_len, head_dim) -> (seq_len, embed_dim)
+                output_i = output_i.transpose(1, 2).reshape(-1, self.embed_dim)
+                outputs.append(output_i)
+            attn_output = torch.cat(outputs, dim=0)
+
+        attn_output = self.out_proj(attn_output)
+        return attn_output
+
+class Siglip2MLP(nn.Module):
+    def __init__(self, config):
+        super().__init__()
+        self.config = config
+        self.activation_fn = ACT2FN[config.hidden_act]
+        self.fc1 = nn.Linear(config.hidden_size, config.intermediate_size)
+        self.fc2 = nn.Linear(config.intermediate_size, config.hidden_size)
+
+    def forward(self, hidden_states: torch.Tensor) -> torch.Tensor:
+        hidden_states = self.fc1(hidden_states)
+        hidden_states = self.activation_fn(hidden_states)
+        hidden_states = self.fc2(hidden_states)
+        return hidden_states
+
+
+class Siglip2EncoderLayer(nn.Module):
+    def __init__(self, config: Siglip2NavitConfig):
+        super().__init__()
+        self.embed_dim = config.hidden_size
+        self.layer_norm1 = nn.LayerNorm(self.embed_dim, eps=config.layer_norm_eps)
+        self.self_attn = Siglip2Attention(config)
+        self.layer_norm2 = nn.LayerNorm(self.embed_dim, eps=config.layer_norm_eps)
+        self.mlp = Siglip2MLP(config)
+
+    def forward(
+        self,
+        hidden_states: torch.Tensor,
+        cu_seqlens: torch.Tensor, 
+        position_embeddings: torch.Tensor
+    ) -> tuple[torch.FloatTensor]:
+        """
+        Args:
+            hidden_states (`torch.FloatTensor`):
+                Input to the layer of shape `(batch, seq_len, embed_dim)`.
+            attention_mask (`torch.FloatTensor`):
+                Attention mask of shape `(batch, 1, q_len, k_v_seq_len)` where padding elements are indicated by very large negative values.
+            output_attentions (`bool`, *optional*, defaults to `False`):
+                Whether or not to return the attentions tensors of all attention layers. See `attentions` under
+                returned tensors for more detail.
+        """
+        residual = hidden_states
+
+        hidden_states = self.layer_norm1(hidden_states)
+        hidden_states = self.self_attn(
+            hidden_states=hidden_states,
+            cu_seqlens=cu_seqlens, 
+            position_embeddings=position_embeddings
+        )
+        hidden_states = residual + hidden_states
+
+        residual = hidden_states
+        hidden_states = self.layer_norm2(hidden_states)
+        hidden_states = self.mlp(hidden_states)
+        hidden_states = residual + hidden_states
+
+        return hidden_states
+
+class Siglip2Encoder(nn.Module):
+    """
+    Transformer encoder consisting of `config.num_hidden_layers` self attention layers. Each layer is a
+    [`Siglip2EncoderLayer`].
+    Args:
+        config: Siglip2NavitConfig
+    """
+
+    def __init__(self, config: Siglip2NavitConfig):
+        super().__init__()
+        self.config = config
+        self.layers = nn.ModuleList([Siglip2EncoderLayer(config) for _ in range(config.num_hidden_layers)])
+        self.gradient_checkpointing = False
+
+        self.rotary_pos_emb = VisionRotaryEmbedding(config.hidden_size // config.num_attention_heads // 2)
+        self.patch_size = config.patch_size
+        self.hidden_stride = config.hidden_stride
+        self.window_size = config.window_size
+        self.spatial_merge_unit = config.hidden_stride * config.hidden_stride
+        self.fullatt_block_indexes = None if config.fullatt_block_indexes is None else [int(i) for i in config.fullatt_block_indexes.split('|')]
+        
+
+    # copied from qwen2.5_vl
+    def rot_pos_emb(self, grid_thw):
+        pos_ids = []
+        for t, h, w in grid_thw:
+            hpos_ids = torch.arange(h).unsqueeze(1).expand(-1, w)
+            hpos_ids = hpos_ids.reshape(
+                h // self.hidden_stride,
+                self.hidden_stride,
+                w // self.hidden_stride,
+                self.hidden_stride,
+            )
+            hpos_ids = hpos_ids.permute(0, 2, 1, 3)
+            hpos_ids = hpos_ids.flatten()
+
+            wpos_ids = torch.arange(w).unsqueeze(0).expand(h, -1)
+            wpos_ids = wpos_ids.reshape(
+                h // self.hidden_stride,
+                self.hidden_stride,
+                w // self.hidden_stride,
+                self.hidden_stride,
+            )
+            wpos_ids = wpos_ids.permute(0, 2, 1, 3)
+            wpos_ids = wpos_ids.flatten()
+            pos_ids.append(torch.stack([hpos_ids, wpos_ids], dim=-1).repeat(t, 1))
+        pos_ids = torch.cat(pos_ids, dim=0)
+        max_grid_size = grid_thw[:, 1:].max()
+        rotary_pos_emb_full = self.rotary_pos_emb(max_grid_size)
+        rotary_pos_emb = rotary_pos_emb_full[pos_ids].flatten(1)
+        return rotary_pos_emb
+
+    def get_window_index(self, grid_thw):
+        window_index: list = []
+        cu_window_seqlens: list = [0]
+        window_index_id = 0
+        vit_merger_window_size = self.window_size // self.hidden_stride // self.patch_size  # patch (after merge) number in each window
+
+        for grid_t, grid_h, grid_w in grid_thw:
+            llm_grid_h, llm_grid_w = (
+                grid_h // self.hidden_stride,  # number of patch after merge
+                grid_w // self.hidden_stride,
+            )
+            index = torch.arange(grid_t * llm_grid_h * llm_grid_w).reshape(grid_t, llm_grid_h, llm_grid_w)
+            pad_h = vit_merger_window_size - llm_grid_h % vit_merger_window_size
+            pad_w = vit_merger_window_size - llm_grid_w % vit_merger_window_size
+            num_windows_h = (llm_grid_h + pad_h) // vit_merger_window_size
+            num_windows_w = (llm_grid_w + pad_w) // vit_merger_window_size
+            index_padded = F.pad(index, (0, pad_w, 0, pad_h), "constant", -100)
+            index_padded = index_padded.reshape(
+                grid_t,
+                num_windows_h,
+                vit_merger_window_size,
+                num_windows_w,
+                vit_merger_window_size,
+            )
+            index_padded = index_padded.permute(0, 1, 3, 2, 4).reshape(
+                grid_t,
+                num_windows_h * num_windows_w,
+                vit_merger_window_size,
+                vit_merger_window_size,
+            )
+            seqlens = (index_padded != -100).sum([2, 3]).reshape(-1)
+            index_padded = index_padded.reshape(-1)
+            index_new = index_padded[index_padded != -100]
+            window_index.append(index_new + window_index_id)
+            cu_seqlens_tmp = seqlens.cumsum(0) * self.spatial_merge_unit + cu_window_seqlens[-1]
+            cu_window_seqlens.extend(cu_seqlens_tmp.tolist())
+            window_index_id += (grid_t * llm_grid_h * llm_grid_w).item()
+        window_index = torch.cat(window_index, dim=0)
+
+        return window_index, cu_window_seqlens
+
+    # Ignore copy
+    def forward(
+        self,
+        inputs_embeds,
+        grid_thws: torch.Tensor,
+        output_hidden_states: bool = False,
+    ) -> Tuple[torch.Tensor, Optional[Tuple[torch.Tensor, ...]]]:
+        r"""
+        Args:
+            inputs_embeds (`torch.FloatTensor` of shape `(batch_size, sequence_length, hidden_size)`):
+                Optionally, instead of passing `input_ids` you can choose to directly pass an embedded representation.
+                This is useful if you want more control over how to convert `input_ids` indices into associated vectors
+                than the model's internal embedding lookup matrix.
+            attention_mask (`torch.Tensor` of shape `(batch_size, sequence_length)`, *optional*):
+                Mask to avoid performing attention on padding token indices. Mask values selected in `[0, 1]`:
+                - 1 for tokens that are **not masked**,
+                - 0 for tokens that are **masked**.
+                [What are attention masks?](../glossary#attention-mask)
+            output_attentions (`bool`, *optional*):
+                Whether or not to return the attentions tensors of all attention layers. See `attentions` under
+                returned tensors for more detail.
+            output_hidden_states (`bool`, *optional*):
+                Whether or not to return the hidden states of all layers. See `hidden_states` under returned tensors
+                for more detail.
+            return_dict (`bool`, *optional*):
+                Whether or not to return a [`~utils.ModelOutput`] instead of a plain tuple.
+        """
+
+        rotary_pos_emb = self.rot_pos_emb(grid_thws)
+        window_index, cu_window_seqlens = self.get_window_index(grid_thws)
+        cu_window_seqlens = torch.tensor(
+            cu_window_seqlens,
+            device=inputs_embeds.device,
+            dtype=grid_thws.dtype if torch.jit.is_tracing() else torch.int32,
+        )
+        cu_window_seqlens = torch.unique_consecutive(cu_window_seqlens)
+
+        seq_len, _ = inputs_embeds.size()
+        inputs_embeds = inputs_embeds.reshape(seq_len // self.spatial_merge_unit, self.spatial_merge_unit, -1)
+        inputs_embeds = inputs_embeds[window_index, :, :]
+        inputs_embeds = inputs_embeds.reshape(seq_len, -1)
+        rotary_pos_emb = rotary_pos_emb.reshape(seq_len // self.spatial_merge_unit, self.spatial_merge_unit, -1)
+        rotary_pos_emb = rotary_pos_emb[window_index, :, :]
+        rotary_pos_emb = rotary_pos_emb.reshape(seq_len, -1)
+        emb = torch.cat((rotary_pos_emb, rotary_pos_emb), dim=-1)
+        position_embeddings = (emb.cos(), emb.sin())
+
+        cu_seqlens = torch.repeat_interleave(grid_thws[:, 1] * grid_thws[:, 2], grid_thws[:, 0]).cumsum(
+            dim=0,
+            # Select dtype based on the following factors:
+            #  - FA2 requires that cu_seqlens_q must have dtype int32
+            #  - torch.onnx.export requires that cu_seqlens_q must have same dtype as grid_thw
+            # See https://github.com/huggingface/transformers/pull/34852 for more information
+            dtype=grid_thws.dtype if torch.jit.is_tracing() else torch.int32,
+        )
+        cu_seqlens = F.pad(cu_seqlens, (1, 0), value=0)
+
+        reverse_indices = torch.argsort(window_index)
+        encoder_states = () if output_hidden_states else None
+
+        hidden_states = inputs_embeds
+        for index, block in enumerate(self.layers):
+            if self.fullatt_block_indexes is None or index in self.fullatt_block_indexes:
+                cu_seqlens_tmp = cu_seqlens
+            else:
+                cu_seqlens_tmp = cu_window_seqlens
+            if self.gradient_checkpointing and self.training:
+                hidden_states = self._gradient_checkpointing_func(block.__call__, hidden_states, cu_seqlens_tmp, position_embeddings)
+            else:
+                hidden_states = block(hidden_states, cu_seqlens_tmp, position_embeddings)
+            if output_hidden_states:
+                hidden_states_ = hidden_states.reshape(seq_len // self.spatial_merge_unit, self.spatial_merge_unit, -1)
+                encoder_states += (hidden_states_[reverse_indices, :].reshape(seq_len, -1),)
+        # tokens = self.post_trunk_norm(tokens)
+        hidden_states = hidden_states.reshape(seq_len // self.spatial_merge_unit, self.spatial_merge_unit, -1)
+        hidden_states = hidden_states[reverse_indices, :].reshape(seq_len, -1)
+
+        return hidden_states, encoder_states
+
+class Siglip2VisionTransformer(nn.Module):
+    def __init__(self, config: Siglip2NavitConfig):
+        super().__init__()
+        self.config = config
+        embed_dim = config.hidden_size
+
+        self.embeddings = Siglip2VisionEmbeddings(config)
+        self.encoder = Siglip2Encoder(config)
+        self.post_layernorm = nn.LayerNorm(embed_dim, eps=config.layer_norm_eps)
+        self._use_flash_attention_2 = config._attn_implementation == "flash_attention_2"
+
+    def forward(
+        self,
+        pixel_values: torch.FloatTensor,
+        grid_thws: torch.LongTensor,
+        output_hidden_states: Optional[bool] = True,
+        return_dict: Optional[bool] = True,
+    ) -> Union[
+        Tuple[torch.Tensor],
+        Tuple[torch.Tensor, Tuple[torch.Tensor, ...]],
+        BaseModelOutputWithNoAttention,
+    ]:
+        r"""
+        spatial_shapes (`torch.LongTensor` of shape `(batch_size, 2)`):
+            Tensor containing the spatial dimensions (height, width) of the input images.
+        """
+        # output_attentions = output_attentions if output_attentions is not None else self.config.output_attentions
+        # output_hidden_states = (
+        #     output_hidden_states if output_hidden_states is not None else self.config.output_hidden_states
+        # )
+
+        hidden_states = self.embeddings(pixel_values, grid_thws)
+
+        last_hidden_state, hidden_states = self.encoder(hidden_states, grid_thws, output_hidden_states)
+        last_hidden_state = self.post_layernorm(last_hidden_state)
+
+        if not return_dict:
+            output = (last_hidden_state,)
+            output += (hidden_states,) if output_hidden_states else ()
+            return output
+        
+        return BaseModelOutputWithNoAttention(
+                last_hidden_state=last_hidden_state, 
+                hidden_states=hidden_states
+            )
+       
+class Siglip2PreTrainedModel(PreTrainedModel):
+    config_class = Siglip2NavitConfig
+    base_model_prefix = "siglip2_navit"
+    supports_gradient_checkpointing = True
+
+    _no_split_modules = [
+        "Siglip2VisionEmbeddings",
+        "Siglip2EncoderLayer",
+    ]
+    _supports_flash_attn_2 = True
+    _supports_sdpa = False
+    _supports_flex_attn = False
+    _supports_attention_backend = True
+
+
+class Siglip2NavitModel(Siglip2PreTrainedModel):
+    config_class = Siglip2NavitConfig
+    main_input_name = "pixel_values"
+
+    def __init__(self, config: Siglip2NavitConfig):
+        super().__init__(config)
+
+        self.vision_model = Siglip2VisionTransformer(config)
+
+    def get_input_embeddings(self) -> nn.Module:
+        return self.vision_model.embeddings.patch_embedding
+
+    def forward(
+        self,
+        pixel_values: torch.FloatTensor,
+        grid_thws: torch.LongTensor,
+        output_hidden_states: Optional[bool] = None,
+        return_dict: Optional[bool] = None,
+    ) -> Union[
+        Tuple[torch.Tensor],
+        Tuple[torch.Tensor, Tuple[torch.Tensor, ...]],
+        BaseModelOutputWithNoAttention,
+    ]:
+        
+        if output_hidden_states is None:
+            output_hidden_states = self.config.output_hidden_states
+        if return_dict is None:
+            return_dict = self.config.use_return_dict
+
+        return self.vision_model(
+            pixel_values=pixel_values,
+            grid_thws=grid_thws,
+            output_hidden_states=output_hidden_states,
+            return_dict=return_dict,
+        )
+
+class VisualEmbedding(torch.nn.Embedding):
+    """
+    A visual embedding layer that can handle both discrete token IDs (long) and continuous
+    soft-token probabilities (float).
+    """
+
+    def forward(self, visual_tokens: Tensor) -> Tensor:
+        if visual_tokens.dtype in [torch.int8, torch.int16, torch.int32, torch.int64, torch.long]:
+            return super().forward(visual_tokens)
+        # Handle soft tokens (probabilities) by matrix multiplication with the embedding weight
+        return torch.matmul(visual_tokens, self.weight)
+
+
+class VisualTokenizer(torch.nn.Module):
+    """
+    Tokenizes images or videos into a sequence of continuous visual tokens.
+    """
+
+    def __init__(self, vit, visual_vocab_size, image_processor_name_or_path, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        self.vit = vit
+        self.image_processor = AutoImageProcessor.from_pretrained(image_processor_name_or_path, do_center_crop=False)
+        head_dim = visual_vocab_size - len(INDICATOR_IDS)
+        self.head = torch.nn.Sequential(
+            torch.nn.Linear(self.vit.config.hidden_size * self.vit.config.hidden_stride ** 2, head_dim, bias=False),
+            torch.nn.LayerNorm(head_dim)
+        )
+
+    def _encode(self, pixel_values, grid_thws):
+        output = self.vit(pixel_values, grid_thws, output_hidden_states=True, return_dict=True)
+        features = output.hidden_states[-1]
+        seq_len, _ = features.shape
+        features = features.reshape(seq_len // (self.vit.config.hidden_stride ** 2), -1)
+        return features
+
+    # Adapted from qwen2_vl
+    @staticmethod
+    def smart_resize(
+        height: int, width: int, factor: int = 28, min_pixels: int = 448 * 448, max_pixels: int = 1344 * 1792
+    ):
+        """Rescales the image so that the following conditions are met:
+        1. Both dimensions are divisible by 'factor'.
+        2. The total number of pixels is within ['min_pixels', 'max_pixels'].
+        3. The aspect ratio is maintained as closely as possible.
+        """
+        if height < factor or width < factor:
+            if height < width:
+                width = round(factor / height * width)
+                height = factor
+            else:
+                height = round(factor / width * height)
+                width = factor
+
+        elif max(height, width) / min(height, width) > 200:
+            if height > width:
+                height = 200 * width
+            else:
+                width = 200 * height
+
+        h_bar = round(height / factor) * factor
+        w_bar = round(width / factor) * factor
+        if h_bar * w_bar > max_pixels:
+            beta = math.sqrt((height * width) / max_pixels)
+            h_bar = math.floor(height / beta / factor) * factor
+            w_bar = math.floor(width / beta / factor) * factor
+        elif h_bar * w_bar < min_pixels:
+            beta = math.sqrt(min_pixels / (height * width))
+            h_bar = math.ceil(height * beta / factor) * factor
+            w_bar = math.ceil(width * beta / factor) * factor
+        return h_bar, w_bar
+
+    def preprocess(
+        self,
+        image: Optional[PIL.Image.Image] = None,
+        video: Optional[List[PIL.Image.Image]] = None,
+        min_pixels: Optional[int] = None,
+        max_pixels: Optional[int] = None
+    ):
+        patch_size = self.vit.config.patch_size
+        temporal_patch_size = self.vit.config.temporal_patch_size
+        hidden_stride = self.vit.config.hidden_stride
+        assert (image is None) ^ (video is None), "Invalid input: expect either image or video"
+        if image is not None:
+            images = [image]
+        else:
+            images = video
+        images = [image.convert("RGB") if image.mode != 'RGB' else image for image in images]
+        width, height = images[0].size
+        processed_images = []
+        for image in images:
+            resized_height, resized_width = self.smart_resize(
+                height,
+                width,
+                factor=patch_size * hidden_stride,
+                min_pixels=min_pixels,
+                max_pixels=max_pixels,
+            )
+            new_size = dict(height=resized_height, width=resized_width)
+            new_image = self.image_processor.preprocess(image, size=new_size, return_tensors="np")['pixel_values'][0]
+            processed_images.append(new_image)
+
+        patches = np.array(processed_images)
+        if patches.shape[0] % temporal_patch_size != 0:
+            repeats = np.repeat(patches[-1][np.newaxis], temporal_patch_size - 1, axis=0)
+            patches = np.concatenate([patches, repeats], axis=0)
+        channel = patches.shape[1]
+        grid_t = patches.shape[0] // temporal_patch_size
+        grid_h, grid_w = resized_height // patch_size, resized_width // patch_size
+        grid_thw = torch.tensor([[grid_t, grid_h, grid_w]])
+
+        patches = patches.reshape(
+            grid_t, temporal_patch_size, channel,
+            grid_h // hidden_stride, hidden_stride, patch_size,
+            grid_w // hidden_stride, hidden_stride, patch_size,
+        )
+        patches = patches.transpose(0, 3, 6, 4, 7, 2, 1, 5, 8)
+        flatten_patches = patches.reshape(
+            grid_t * grid_h * grid_w, channel * temporal_patch_size * patch_size * patch_size
+        )
+        flatten_patches = torch.tensor(flatten_patches)
+
+        return flatten_patches, grid_thw
+
+    def forward(
+        self, pixel_values, grid_thws
+    ) -> torch.Tensor:  # [BatchSize, ImageShape] -> [BatchSize, #Token, VocabSize]
+        features = self._encode(pixel_values, grid_thws)
+        logits = self.head(features)
+        tokens = torch.softmax(logits, dim=-1, dtype=torch.float32).to(logits.dtype)
+
+        token_len, _ = tokens.shape
+        padding_tensor = torch.zeros(size=(token_len, len(INDICATOR_IDS)),
+                                     dtype=tokens.dtype,
+                                     device=tokens.device,
+                                     layout=tokens.layout,
+                                     requires_grad=False)
+        tokens = torch.cat((tokens, padding_tensor), dim=1)
+        return tokens
+
+
+class OvisPreTrainedModel(PreTrainedModel):
+    config_class = Ovis2_5_Config
+    base_model_prefix = "ovis2_5"
+
+
+class Ovis2_5(OvisPreTrainedModel):
+    _supports_flash_attn_2 = True
+
+    def __init__(self, config: Ovis2_5_Config, *inputs, **kwargs):
+        super().__init__(config, *inputs, **kwargs)
+
+        self.llm = AutoModelForCausalLM.from_config(self.config.llm_config)
+        assert self.config.hidden_size == self.llm.config.hidden_size, "hidden size mismatch"
+        self.text_tokenizer = AutoTokenizer.from_pretrained(self.config.name_or_path)
+        self.visual_tokenizer = VisualTokenizer(vit=AutoModel.from_config(self.config.vit_config),
+                                                visual_vocab_size=self.config.visual_vocab_size,
+                                                image_processor_name_or_path=self.config.name_or_path)
+
+        self.vte = VisualEmbedding(self.config.visual_vocab_size, self.config.hidden_size,
+                                   device=self.visual_tokenizer.vit.device, dtype=self.visual_tokenizer.vit.dtype)
+        indicator_token_indices = torch.arange(
+            self.config.visual_vocab_size - len(INDICATOR_IDS),
+            self.config.visual_vocab_size,
+            dtype=torch.long
+        )
+        self.register_buffer("indicator_token_indices", indicator_token_indices, persistent=False)
+
+        def _merge_modules(modules_list: tuple):
+            merged_modules = []
+            for modules in modules_list:
+                merged_modules.extend(modules if modules else [])
+            return merged_modules
+
+        # Standard model configurations for parallelism and device placement
+        self._no_split_modules = _merge_modules(
+            (self.llm._no_split_modules, self.visual_tokenizer.vit._no_split_modules))
+        self._skip_keys_device_placement = self.llm._skip_keys_device_placement
+        self._keep_in_fp32_modules = _merge_modules(
+            (self.llm._keep_in_fp32_modules, self.visual_tokenizer.vit._keep_in_fp32_modules))
+        self.is_parallelizable = all((self.llm.is_parallelizable, self.visual_tokenizer.vit.is_parallelizable))
+        self.supports_gradient_checkpointing = True
+
+    def tie_weights(self):
+        self.llm.tie_weights()
+
+    def get_wte(self):
+        return self.llm.get_input_embeddings()
+
+    def forward(
+        self,
+        input_ids: torch.Tensor,
+        attention_mask: torch.Tensor,
+        pixel_values: Optional[torch.Tensor],
+        grid_thws: Optional[torch.Tensor],
+        labels: Optional[torch.Tensor] = None,
+        **kwargs
+    ):
+        inputs_embeds = self.merge_multimodal(
+            input_ids=input_ids,
+            pixel_values=pixel_values,
+            grid_thws=grid_thws,
+        )
+        return self.llm(inputs_embeds=inputs_embeds, attention_mask=attention_mask, labels=labels, **kwargs)
+
+    def merge_multimodal(
+        self,
+        input_ids: torch.Tensor,
+        pixel_values: Optional[torch.Tensor],
+        grid_thws: Optional[torch.Tensor],
+    ):
+        placeholder_token_mask = torch.lt(input_ids, 0)
+        multimodal_embeds = self.get_wte()(torch.masked_fill(input_ids, placeholder_token_mask, 0))
+
+        if pixel_values is not None:
+            visual_indicator_embeds = self.vte(self.indicator_token_indices).to(
+                dtype=multimodal_embeds.dtype, device=multimodal_embeds.device
+            )
+            visual_tokens = self.visual_tokenizer(pixel_values, grid_thws)
+            visual_embeds = self.vte(visual_tokens).to(dtype=multimodal_embeds.dtype, device=multimodal_embeds.device)
+
+            for i, indicator_id in enumerate(INDICATOR_IDS):
+                multimodal_embeds[input_ids == indicator_id] = visual_indicator_embeds[i]
+            multimodal_embeds[input_ids == VISUAL_ATOM_ID] = visual_embeds
+
+        return multimodal_embeds
+
+    def _merge_inputs(
+        self, raw_input_ids, placeholder_id, grid_thws, indicator_begin_id, indicator_end_id
+    ):
+        input_ids = []
+        prev_index = 0
+        placeholder_indexes = [i for i, v in enumerate(raw_input_ids) if v == placeholder_id]
+        for placeholder_index, grid_thw in zip(placeholder_indexes, grid_thws):
+            input_ids.extend(raw_input_ids[prev_index:placeholder_index])
+            num_image_atoms = grid_thw.prod().item()
+            num_image_atoms //= self.visual_tokenizer.vit.config.hidden_stride ** 2
+            num_image_atoms //= self.visual_tokenizer.vit.config.temporal_patch_size
+            input_ids.extend([indicator_begin_id] + [VISUAL_ATOM_ID] * num_image_atoms + [indicator_end_id])
+            prev_index = placeholder_index + 1
+        input_ids.extend(raw_input_ids[prev_index:])
+        return input_ids
+
+    def _tokenize_with_visual_placeholder(self, text):
+        placeholder = VIDEO_PLACEHOLDER if VIDEO_PLACEHOLDER in text else IMAGE_PLACEHOLDER
+        placeholder_id = VIDEO_PLACEHOLDER_ID if VIDEO_PLACEHOLDER in text else IMAGE_PLACEHOLDER_ID
+        chunks = [self.text_tokenizer(chunk, add_special_tokens=False).input_ids for chunk in text.split(placeholder)]
+        input_ids = chunks[0]
+        for chunk in chunks[1:]:
+            input_ids.append(placeholder_id)
+            input_ids.extend(chunk)
+        return input_ids
+
+    def preprocess_inputs(
+        self,
+        messages: List[Union[str, Dict]],
+        min_pixels=448 * 448,
+        max_pixels=1792 * 1792,
+        add_generation_prompt=True,
+        enable_thinking=False
+    ):
+        text = self.text_tokenizer.apply_chat_template(
+            messages,
+            tokenize=False,
+            add_generation_prompt=add_generation_prompt,
+            enable_thinking=enable_thinking
+        )
+        input_ids = self._tokenize_with_visual_placeholder(text)
+        images = []
+        videos = []
+        for message in messages:
+            content = message["content"]
+            if isinstance(content, list):
+                images.extend([item["image"] for item in content if item.get("image") is not None])
+                videos.extend([item["video"] for item in content if item.get("video") is not None])
+        if images and videos:
+            raise ValueError(
+                "Multiple visual input data types detected (both image and video provided). "
+                "This model supports only one type of visual input data at a time. "
+                "Please provide either image or video, but not both."
+            )
+
+        pixel_values, grid_thws = None, None
+        if images:
+            pixel_values, grid_thws = zip(
+                *(self.visual_tokenizer.preprocess(image=image, min_pixels=min_pixels, max_pixels=max_pixels)
+                  for image in images)
+            )
+            input_ids = self._merge_inputs(
+                input_ids, IMAGE_PLACEHOLDER_ID, grid_thws, INDICATOR_IDS[0], INDICATOR_IDS[1]
+            )
+            pixel_values = torch.cat(pixel_values, dim=0)
+            grid_thws = torch.cat(grid_thws, dim=0)
+        elif videos:
+            assert len(videos) == 1, "only support single video"
+            pixel_values, grid_thws = self.visual_tokenizer.preprocess(
+                video=videos[0], min_pixels=min_pixels, max_pixels=max_pixels
+            )
+            input_ids = self._merge_inputs(
+                input_ids, VIDEO_PLACEHOLDER_ID, grid_thws, INDICATOR_IDS[2], INDICATOR_IDS[3]
+            )
+
+        input_ids = torch.tensor(input_ids, dtype=torch.long).unsqueeze(0)
+
+        return input_ids, pixel_values, grid_thws
+
+    def generate(
+        self,
+        inputs: Optional[torch.Tensor] = None,
+        **kwargs,
+    ) -> Union[GenerateOutput, torch.LongTensor]:
+        attention_mask = torch.ne(inputs, self.text_tokenizer.pad_token_id).to(device=inputs.device)
+        inputs_embeds = self.merge_multimodal(
+            input_ids=inputs,
+            pixel_values=kwargs.pop('pixel_values', None),
+            grid_thws=kwargs.pop('grid_thws', None)
+        )
+        enable_thinking = kwargs.pop('enable_thinking', False)
+        enable_thinking_budget = kwargs.pop('enable_thinking_budget', False)
+        thinking_budget = kwargs.pop('thinking_budget', 1024)
+        
+        if enable_thinking and enable_thinking_budget:
+            actual_max_new_tokens = kwargs['max_new_tokens']
+            kwargs['max_new_tokens'] = thinking_budget
+            generated_ids = self.llm.generate(inputs=None, inputs_embeds=inputs_embeds, attention_mask=attention_mask, **kwargs)
+            output_ids = generated_ids
+            output_ids_list = generated_ids[0]
+            
+            # check if the generation has already finished (151645 is <|im_end|>)
+            if 151645 not in output_ids_list:
+                # check if the thinking process has finished (151668 is </think>)
+                # and prepare the second model input
+                if 151668 not in output_ids_list:
+                    early_stopping_text = "\n\nConsidering the limited time by the user, I have to give the solution based on the thinking directly now.\n</think>\n\n"
+                    early_stopping_ids = self.text_tokenizer(early_stopping_text, return_tensors="pt", return_attention_mask=False).input_ids.to(inputs.device)
+                    input_ids_appendent = torch.cat([output_ids, early_stopping_ids], dim=-1)
+                    kwargs['streamer'].put(early_stopping_ids) if 'streamer' in kwargs else None
+                else:
+                    input_ids_appendent = output_ids
+                
+
+                # second generation
+                new_inputs = torch.cat([inputs, input_ids_appendent], dim=-1)
+                attention_mask = torch.ne(new_inputs, self.text_tokenizer.pad_token_id).to(device=inputs.device)
+                inputs_embeds_appendent = self.merge_multimodal(
+                    input_ids=input_ids_appendent,
+                    pixel_values=None,
+                    grid_thws=None
+                )
+                new_inputs_embeds = torch.cat([inputs_embeds, inputs_embeds_appendent], dim=-2)
+                
+                kwargs['max_new_tokens'] = inputs_embeds.size(-2) + actual_max_new_tokens - new_inputs_embeds.size(-2)
+                generated_ids2 = self.llm.generate(inputs=None, inputs_embeds=new_inputs_embeds, attention_mask=attention_mask, **kwargs)
+                kwargs['streamer'].manual_end() if 'streamer' in kwargs else None
+                return torch.cat([input_ids_appendent, generated_ids2], dim=-1)
+            
+            else:
+                kwargs['streamer'].manual_end() if 'streamer' in kwargs else None
+                return generated_ids
+            
+        else:
+            generated_ids = self.llm.generate(inputs=None, inputs_embeds=inputs_embeds, attention_mask=attention_mask, **kwargs)
+            kwargs['streamer'].manual_end() if 'streamer' in kwargs else None
+            return generated_ids
+
+
+AutoConfig.register('siglip2_navit', Siglip2NavitConfig)
+AutoModel.register(Siglip2NavitConfig, Siglip2NavitModel)
+AutoConfig.register("ovis2_5", Ovis2_5_Config)
+AutoModelForCausalLM.register(Ovis2_5_Config, Ovis2_5)

preprocessor_config.json

@@ -0,0 +1,24 @@
+{
+  "do_convert_rgb": null,
+  "do_normalize": true,
+  "do_rescale": true,
+  "do_resize": true,
+  "image_mean": [
+    0.5,
+    0.5,
+    0.5
+  ],
+  "image_processor_type": "SiglipImageProcessor",
+  "image_std": [
+    0.5,
+    0.5,
+    0.5
+  ],
+  "processor_class": "SiglipProcessor",
+  "resample": 2,
+  "rescale_factor": 0.00392156862745098,
+  "size": {
+    "height": 512,
+    "width": 512
+  }
+}

special_tokens_map.json

@@ -0,0 +1,31 @@
+{
+  "additional_special_tokens": [
+    "<|im_start|>",
+    "<|im_end|>",
+    "<|object_ref_start|>",
+    "<|object_ref_end|>",
+    "<|box_start|>",
+    "<|box_end|>",
+    "<|quad_start|>",
+    "<|quad_end|>",
+    "<|vision_start|>",
+    "<|vision_end|>",
+    "<|vision_pad|>",
+    "<|image_pad|>",
+    "<|video_pad|>"
+  ],
+  "eos_token": {
+    "content": "<|im_end|>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "pad_token": {
+    "content": "<|endoftext|>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  }
+}

tokenizer_config.json

@@ -0,0 +1,247 @@
+{
+  "add_bos_token": false,
+  "add_prefix_space": false,
+  "added_tokens_decoder": {
+    "151643": {
+      "content": "<|endoftext|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151644": {
+      "content": "<|im_start|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151645": {
+      "content": "<|im_end|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151646": {
+      "content": "<|object_ref_start|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151647": {
+      "content": "<|object_ref_end|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151648": {
+      "content": "<|box_start|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151649": {
+      "content": "<|box_end|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151650": {
+      "content": "<|quad_start|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151651": {
+      "content": "<|quad_end|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151652": {
+      "content": "<|vision_start|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151653": {
+      "content": "<|vision_end|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151654": {
+      "content": "<|vision_pad|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151655": {
+      "content": "<|image_pad|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151656": {
+      "content": "<|video_pad|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151657": {
+      "content": "<tool_call>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151658": {
+      "content": "</tool_call>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151659": {
+      "content": "<|fim_prefix|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151660": {
+      "content": "<|fim_middle|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151661": {
+      "content": "<|fim_suffix|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151662": {
+      "content": "<|fim_pad|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151663": {
+      "content": "<|repo_name|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151664": {
+      "content": "<|file_sep|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151665": {
+      "content": "<tool_response>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151666": {
+      "content": "</tool_response>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151667": {
+      "content": "<think>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151668": {
+      "content": "</think>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    }
+  },
+  "additional_special_tokens": [
+    "<|im_start|>",
+    "<|im_end|>",
+    "<|object_ref_start|>",
+    "<|object_ref_end|>",
+    "<|box_start|>",
+    "<|box_end|>",
+    "<|quad_start|>",
+    "<|quad_end|>",
+    "<|vision_start|>",
+    "<|vision_end|>",
+    "<|vision_pad|>",
+    "<|image_pad|>",
+    "<|video_pad|>",
+    "<image>",
+    "<video>",
+    "<ovis_visual_atom>",
+    "<ovis_image_start>",
+    "<ovis_image_end>",
+    "<ovis_video_start>",
+    "<ovis_video_end>"
+  ],
+  "bos_token": null,
+  "chat_template": "{%- for message in messages %}{{- '<|im_start|>' + message.role + '\n'}}{%- if message.role == 'system' or message.role == 'user' %}{%- if message.content is string %}{{- message.content | replace('<image>', '') | replace('<video>', '') }}{%- else %}{%- for item in message.content %}{%- if item.type == 'text' and 'text' in item %}{{- item.text | replace('<image>', '') | replace('<video>', '') }}{%- elif item.type == 'image' %}{{- '<image>'}}{%- elif item.type == 'video' %}{{- '<video>'}}{%- else %}{{- raise_exception('Invalid content type. Supported types for system and user are text, image, video.')}}{%- endif %}{%- if not loop.last %}{{- '\n'}}{%- endif %}{%- endfor %}{%- endif %}{%- elif message.role == 'assistant' %}{%- set content = '' %}{%- if message.content is string %}{%- set content = message.content | replace('<image>', '') | replace('<video>', '') %}{%- else %}{%- set ns = namespace(content='') -%}{%- for item in message.content %}{%- if item.type == 'text' and 'text' in item %}{%- set ns.content = ns.content ~ (item.text | replace('<image>', '') | replace('<video>', '')) %}{%- else %}{{- raise_exception('Invalid content type. Supported type for assistant is text.')}}{%- endif %}{%- endfor %}{%- set content = ns.content -%}{%- endif %}{%- set content = content.split('</think>')[-1].lstrip('\n') %}{{- content }}{%- else %}{{- raise_exception('Invalid role. Supported roles are system, user, assistant.')}}{%- endif %}{{- '<|im_end|>\n'}}{%- endfor %}{%- if add_generation_prompt %}{{- '<|im_start|>assistant\n' }}{%- if enable_thinking is defined and enable_thinking is false %}{{- '<think>\n\n</think>\n\n' }}{%- endif %}{%- endif %}",
+  "clean_up_tokenization_spaces": false,
+  "eos_token": "<|im_end|>",
+  "errors": "replace",
+  "extra_special_tokens": {},
+  "model_max_length": 131072,
+  "pad_token": "<|endoftext|>",
+  "split_special_tokens": false,
+  "tokenizer_class": "Qwen2Tokenizer",
+  "unk_token": null
+}