Upload folder using huggingface_hub

.gitattributes

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

README.md

@@ -0,0 +1,212 @@
+---
+base_model:
+  - black-forest-labs/FLUX.2-dev
+library_name: diffusers
+license: apache-2.0
+pipeline_tag: image-to-image
+tags:
+  - diffusers
+  - safetensors
+  - image-editing
+  - vbvr
+datasets:
+  - Video-Reason/VBVR-Dataset
+---
+
+# VBVR-Pro: A Scalable and Verifiable Suite for Native Visual Reasoning
+
+<a href="https://video-reason.com/?v=pro" target="_blank">
+    <img alt="Project Page" src="https://img.shields.io/badge/Project%20-%20Homepage-4285F4" height="20" />
+</a>
+<a href="https://github.com/Video-Reason/VBVR-Pro-Bench" target="_blank">
+    <img alt="Code" src="https://img.shields.io/badge/Evaluation_code-VBVR_Pro_Bench-100000?style=flat-square&logo=github&logoColor=white" height="20" />
+</a>
+<a href="https://github.com/Video-Reason/VBVR-Pro" target="_blank">
+    <img alt="Code" src="https://img.shields.io/badge/Training_Inferenceing-VBVR_Pro-100000?style=flat-square&logo=github&logoColor=white" height="20" />
+</a>
+<a href="https://huggingface.co/papers/2602.20159" target="_blank">
+    <img alt="arXiv" src="https://img.shields.io/badge/arXiv-VBVR_Pro-red?logo=arxiv" height="20" />
+</a>
+<a href="https://huggingface.co/datasets/Video-Reason/VBVR-Pro-SFT-Video" target="_blank">
+    <img alt="Dataset" src="https://img.shields.io/badge/%F0%9F%A4%97%20_VBVR_Pro_Dataset-Data-ffc107?color=ffc107&logoColor=white" height="20" />
+</a>
+<a href="https://huggingface.co/datasets/Video-Reason/VBVR-Pro-Bench/tree/main" target="_blank">
+    <img alt="Bench Data" src="https://img.shields.io/badge/%F0%9F%A4%97%20_VBVR_Pro_Bench-Data-ffc107?color=ffc107&logoColor=white" height="20" />
+</a>
+<a href="https://video-reason.com/pro/bench/#leaderboard" target="_blank">
+    <img alt="Leaderboard" src="https://img.shields.io/badge/%F0%9F%A4%97%20_VBVR_Pro_Bench-Leaderboard-ffc107?color=ffc107&logoColor=white" height="20" />
+</a>
+
+## Overview
+Native visual reasoning, i.e., reasoning through visual generation, has recently emerged as a promising direction for studying visual intelligence beyond language. Yet progress remains bottlenecked by the lack of scalable training tasks, reliable feedback, and controlled comparisons across generative substrates. In this work, we introduce **VBVR-Pro**, a closed-loop testbed that makes native visual reasoning through generation trainable, verifiable, optimizable, and experimentally controllable. **1) Task scaling.** VBVR-Pro turns visual reasoning into a controlled task space of *300* procedurally generated tasks. Models trained on VBVR-Pro show strong transfer beyond the proposed suite across *six* held-out visual reasoning benchmarks such as RISE-Video, MME-CoF-Pro, and BabyVision. Further analysis validates that these gains reflect visual reasoning rather than instruction-pattern fitting. **2) Verifiable rewards.** VBVR-Pro provides verifiable reward scorers for task-grounded evaluation. Through a systematic study of leading MLLMs as judges, we identify recurring failure modes of the prevalent *VLM-as-a-judge* paradigm. In contrast, the proposed scorers are grounded on verifiable task-specific rules, achieve fine-grained alignment with human judgments. Importantly, they serve as reliable reward signals for large-scale multi-task reinforcement learning and demonstrate stronger post-RL performance across visual reasoning tasks. **3) Mechanism study.** VBVR-Pro enables controlled modality studies across more than *30* image, video, and interleaved generators. Our analysis shows that video generation remains strongest for tasks requiring persistent spatiotemporal state tracking, while interleaved generation provides a compute-efficient alternative by externalizing intermediate visual states. Critically, ablations and probing confirm the presence of vision-native trajectories, that are a more crucial substrate than explicit linguistic chains of thought for visual reasoning. We release all data, models, scorers, and code to facilitate future research.
+
+The models are presented in the paper [VBVR-Pro: A Scalable and Verifiable Suite for Native Visual Reasoning](https://huggingface.co/papers/2602.20159).
+
+## Models Zoo
+<table border="1" cellspacing="0" cellpadding="4" style="border-collapse: collapse; width: 100%;">
+  <thead>
+    <tr>
+      <th width="260" style="min-width: 260px;">Model</th>
+      <th>Base Architecture</th>
+      <th>Other Remarks</th>
+    </tr>
+  </thead>
+  <tbody>
+    <tr style="border-top: 4px solid #6b7280; background-color: #e5e7eb;"><th colspan="3" align="left">Image Generation Models</th></tr>
+    <tr><td><a href="https://huggingface.co/Video-Reason/VBVR-Pro-BAGEL">VBVR-Pro-BAGEL</a></td><td>BAGEL-7B-MoT</td><td>Complete model</td></tr>
+    <tr><td><a href="https://huggingface.co/Video-Reason/VBVR-Pro-FLUX2-dev"><strong>VBVR-Pro-FLUX2-dev</strong></a></td><td>FLUX.2-dev</td><td>Complete model, Diffusers format</td></tr>
+    <tr><td><a href="https://huggingface.co/Video-Reason/VBVR-Pro-FLUX2-dev-diffsynth">VBVR-Pro-FLUX2-dev-diffsynth</a></td><td>FLUX.2-dev</td><td>LoRA model, DiffSynth format</td></tr>
+    <tr><td><a href="https://huggingface.co/Video-Reason/VBVR-Pro-Qwen-Image-Edit">VBVR-Pro-Qwen-Image-Edit</a></td><td>Qwen-Image-Edit-2511</td><td>Complete model, Diffusers format</td></tr>
+    <tr><td><a href="https://huggingface.co/Video-Reason/VBVR-Pro-Qwen-Image-Edit-diffsynth">VBVR-Pro-Qwen-Image-Edit-diffsynth</a></td><td>Qwen-Image-Edit-2511</td><td>LoRA model, DiffSynth format</td></tr>
+    <tr style="border-top: 4px solid #6b7280; background-color: #e5e7eb;"><th colspan="3" align="left">Interleaved Image Generation Models</th></tr>
+    <tr><td><a href="https://huggingface.co/Video-Reason/VBVR-Pro-ThinkMorph">VBVR-Pro-ThinkMorph</a></td><td>ThinkMorph-7B</td><td>Complete model</td></tr>
+    <tr><td><a href="https://huggingface.co/Video-Reason/VBVR-Pro-SenseNova-U1">VBVR-Pro-SenseNova-U1</a></td><td>SenseNova-U1-8B-MoT</td><td>Complete model</td></tr>
+    <tr style="border-top: 4px solid #6b7280; background-color: #e5e7eb;"><th colspan="3" align="left">Video Generation Models</th></tr>
+    <tr><td><a href="https://huggingface.co/Video-Reason/VBVR-Pro-LTX2.3">VBVR-Pro-LTX2.3</a></td><td>LTX-Video-2.3</td><td>Complete model, Diffusers format</td></tr>
+    <tr><td><a href="https://huggingface.co/Video-Reason/VBVR-Pro-LTX2.3-diffsynth">VBVR-Pro-LTX2.3-diffsynth</a></td><td>LTX-Video-2.3</td><td>LoRA model, DiffSynth format</td></tr>
+    <tr><td><a href="https://huggingface.co/Video-Reason/VBVR-Pro-Wan2.1-I2V-14B">VBVR-Pro-Wan2.1-I2V-14B</a></td><td>Wan2.1-I2V-14B-720P</td><td>Complete model, Diffusers format</td></tr>
+    <tr><td><a href="https://huggingface.co/Video-Reason/VBVR-Pro-Wan2.1-I2V-14B-diffsynth">VBVR-Pro-Wan2.1-I2V-14B-diffsynth</a></td><td>Wan2.1-I2V-14B-720P</td><td>LoRA model, DiffSynth format</td></tr>
+    <tr><td><a href="https://huggingface.co/Video-Reason/VBVR-Pro-Wan2.2-I2V-A14B">VBVR-Pro-Wan2.2-I2V-A14B</a></td><td>Wan2.2-I2V-A14B</td><td>Complete model, Diffusers format</td></tr>
+    <tr><td><a href="https://huggingface.co/Video-Reason/VBVR-Pro-Wan2.2-I2V-A14B-diffsynth">VBVR-Pro-Wan2.2-I2V-A14B-diffsynth</a></td><td>Wan2.2-I2V-A14B</td><td>LoRA model, DiffSynth format</td></tr>
+    <tr><td><a href="https://huggingface.co/Video-Reason/VBVR-Pro-Wan2.2-TI2V-5B">VBVR-Pro-Wan2.2-TI2V-5B</a></td><td>Wan2.2-TI2V-5B</td><td>Complete model, Diffusers format</td></tr>
+    <tr><td><a href="https://huggingface.co/Video-Reason/VBVR-Pro-Wan2.2-TI2V-5B-diffsynth">VBVR-Pro-Wan2.2-TI2V-5B-diffsynth</a></td><td>Wan2.2-TI2V-5B</td><td>LoRA model, DiffSynth format</td></tr>
+  </tbody>
+</table>
+
+## Release Information
+
+VBVR-Pro finetuned **FLUX.2-dev** for instruction-based image editing,
+trained on the
+[VBVR-Dataset](https://huggingface.co/datasets/Video-Reason/VBVR-Dataset).
+
+This is the **merged model** — the LoRA (rank 32) has already been fused into
+the transformer weights. Load it directly with
+[Diffusers](https://github.com/huggingface/diffusers); no extra adapters or
+DiffSynth-Studio needed.
+
+| Format | Repository |
+|---|---|
+| Merged / Diffusers (this repo) | `VBVR-Pro-FLUX2-dev` |
+| DiffSynth LoRA | `VBVR-Pro-FLUX2-dev-diffsynth` |
+
+In this release, we present all models presented in paper 
+[**VBVR-Pro-Trained-Models**](https://huggingface.co/collections/Video-Reason/VBVR-Pro),
+[**VBVR-Pro-Dataset-Video**](https://huggingface.co/datasets/Video-Reason/VBVR-Pro-SFT-Video),
+[**VBVR-Pro-Dataset-Image**](https://huggingface.co/datasets/Video-Reason/VBVR-Pro-SFT-Image),
+[**VBVR-Pro-Bench**](https://huggingface.co/datasets/Video-Reason/VBVR-Pro-Bench),
+[**VBVR-Pro-Code**](https://github.com/Video-Reason/VBVR-Pro) and 
+[**VBVR-Bench-Leaderboard**](https://video-reason.com/pro/bench/#leaderboard).
+
+## VBVR-Pro Benchmark Results
+<table border="1" cellspacing="0" cellpadding="4" style="border-collapse: collapse; width: 100%; font-size: 12px;">
+  <thead>
+    <tr>
+      <th rowspan="2" width="260" style="min-width: 260px;">Models</th>
+      <th rowspan="2">Overall</th>
+      <th colspan="6">In-Domain by Category</th>
+      <th colspan="6">Out-of-Domain by Category</th>
+    </tr>
+    <tr>
+      <th>Avg.</th><th>Abst.</th><th>Know.</th><th>Perc.</th><th>Spat.</th><th>Trans.</th>
+      <th>Avg.</th><th>Abst.</th><th>Know.</th><th>Perc.</th><th>Spat.</th><th>Trans.</th>
+    </tr>
+  </thead>
+  <tbody>
+    <tr style="border-top: 4px solid #6b7280; background-color: #e5e7eb;"><th colspan="14" align="left">Image Generation Models</th></tr>
+    <tr style="background-color: #dbeafe;"><th colspan="14" align="left">Proprietary Models</th></tr>
+    <tr><td>Qwen-Image-2.0</td><td><u>0.313</u></td><td><u>0.248</u></td><td><u>0.269</u></td><td><u>0.196</u></td><td><u>0.225</u></td><td><u>0.170</u></td><td><u>0.132</u></td><td><u>0.378</u></td><td><u>0.341</u></td><td><u>0.235</u></td><td><u>0.391</u></td><td><u>0.384</u></td><td><u>0.080</u></td></tr>
+    <tr><td>Seedream-5.0-Pro</td><td><strong>0.557</strong></td><td><strong>0.485</strong></td><td><strong>0.518</strong></td><td><strong>0.312</strong></td><td><strong>0.509</strong></td><td><strong>0.401</strong></td><td><strong>0.217</strong></td><td><strong>0.629</strong></td><td><strong>0.507</strong></td><td><strong>0.455</strong></td><td><strong>0.661</strong></td><td><strong>0.559</strong></td><td><strong>0.202</strong></td></tr>
+    <tr style="background-color: #dbeafe;"><th colspan="14" align="left">Open-source Models</th></tr>
+    <tr><td>BAGEL-7B-MoT</td><td>0.089</td><td><u>0.066</u></td><td>0.039</td><td><u>0.085</u></td><td>0.067</td><td>0.046</td><td>0.027</td><td>0.111</td><td><strong>0.201</strong></td><td>0.031</td><td>0.073</td><td>0.028</td><td><strong>0.121</strong></td></tr>
+    <tr><td>FLUX.2-dev</td><td><strong>0.157</strong></td><td><strong>0.108</strong></td><td><u>0.088</u></td><td><strong>0.109</strong></td><td><u>0.072</u></td><td><u>0.100</u></td><td><strong>0.066</strong></td><td><strong>0.206</strong></td><td><u>0.197</u></td><td><strong>0.165</strong></td><td><strong>0.184</strong></td><td><strong>0.241</strong></td><td>0.077</td></tr>
+    <tr><td>Qwen-Image-Edit</td><td><u>0.134</u></td><td><strong>0.108</strong></td><td><strong>0.092</strong></td><td>0.082</td><td><strong>0.100</strong></td><td><strong>0.109</strong></td><td><u>0.056</u></td><td><u>0.159</u></td><td>0.176</td><td><u>0.063</u></td><td><u>0.141</u></td><td><u>0.182</u></td><td><u>0.082</u></td></tr>
+    <tr style="background-color: #dbeafe;"><th colspan="14" align="left">Strong Baselines</th></tr>
+    <tr><td>VBVR-Pro-BAGEL</td><td>0.172</td><td>0.168</td><td>0.199</td><td>0.105</td><td>0.110</td><td>0.213</td><td>0.055</td><td>0.176</td><td>0.254</td><td>0.104</td><td>0.148</td><td>0.015</td><td><u>0.145</u></td></tr>
+    <tr><td>VBVR-Pro-FLUX.2</td><td><strong>0.407</strong></td><td><strong>0.484</strong></td><td><strong>0.483</strong></td><td><strong>0.323</strong></td><td><strong>0.367</strong></td><td><strong>0.449</strong></td><td><strong>0.336</strong></td><td><strong>0.330</strong></td><td><strong>0.361</strong></td><td><strong>0.272</strong></td><td><strong>0.255</strong></td><td><strong>0.454</strong></td><td>0.128</td></tr>
+    <tr><td>VBVR-Pro-Qwen-Image</td><td><u>0.322</u></td><td><u>0.332</u></td><td><u>0.298</u></td><td><u>0.217</u></td><td><u>0.193</u></td><td><u>0.431</u></td><td><u>0.222</u></td><td><u>0.311</u></td><td><u>0.341</u></td><td><u>0.239</u></td><td><u>0.233</u></td><td><u>0.413</u></td><td><strong>0.181</strong></td></tr>
+    <tr style="border-top: 4px solid #6b7280; background-color: #e5e7eb;"><th colspan="14" align="left">Interleaved Image Generation Models</th></tr>
+    <tr style="background-color: #dbeafe;"><th colspan="14" align="left">Proprietary Models</th></tr>
+    <tr><td>GPT-Image-2</td><td><u>0.507</u></td><td><u>0.428</u></td><td><u>0.456</u></td><td><u>0.318</u></td><td><u>0.428</u></td><td><u>0.206</u></td><td><strong>0.300</strong></td><td><u>0.587</u></td><td><u>0.398</u></td><td><u>0.413</u></td><td><u>0.633</u></td><td><u>0.480</u></td><td><strong>0.303</strong></td></tr>
+    <tr><td>Nano Banana Pro</td><td><strong>0.564</strong></td><td><strong>0.480</strong></td><td><strong>0.518</strong></td><td><strong>0.422</strong></td><td><strong>0.512</strong></td><td><strong>0.285</strong></td><td><u>0.174</u></td><td><strong>0.648</strong></td><td><strong>0.553</strong></td><td><strong>0.499</strong></td><td><strong>0.657</strong></td><td><strong>0.585</strong></td><td><u>0.220</u></td></tr>
+    <tr style="background-color: #dbeafe;"><th colspan="14" align="left">Open-source Models</th></tr>
+    <tr><td>ThinkMorph-7B</td><td>0.154</td><td>0.113</td><td>0.100</td><td>0.082</td><td>0.101</td><td>0.148</td><td>0.031</td><td>0.195</td><td>0.176</td><td>0.166</td><td>0.163</td><td>0.253</td><td>0.103</td></tr>
+    <tr><td>VBVR-SenseNova-U1</td><td><u>0.408</u></td><td><u>0.469</u></td><td><u>0.356</u></td><td><u>0.313</u></td><td><u>0.373</u></td><td><strong>0.386</strong></td><td><strong>0.477</strong></td><td><u>0.347</u></td><td><u>0.291</u></td><td><u>0.317</u></td><td><u>0.275</u></td><td><u>0.480</u></td><td><u>0.238</u></td></tr>
+    <tr><td>SenseNova-U1-8B-MoT</td><td><strong>0.565</strong></td><td><strong>0.533</strong></td><td><strong>0.501</strong></td><td><strong>0.395</strong></td><td><strong>0.544</strong></td><td><u>0.355</u></td><td><u>0.349</u></td><td><strong>0.597</strong></td><td><strong>0.448</strong></td><td><strong>0.495</strong></td><td><strong>0.533</strong></td><td><strong>0.717</strong></td><td><strong>0.401</strong></td></tr>
+    <tr style="background-color: #dbeafe;"><th colspan="14" align="left">Strong Baselines</th></tr>
+    <tr><td>VBVR-Pro-ThinkMorph</td><td><u>0.373</u></td><td><u>0.402</u></td><td><u>0.403</u></td><td><u>0.344</u></td><td><u>0.238</u></td><td><u>0.454</u></td><td><u>0.184</u></td><td><u>0.344</u></td><td><u>0.367</u></td><td><u>0.224</u></td><td><u>0.238</u></td><td><u>0.535</u></td><td><u>0.257</u></td></tr>
+    <tr><td>VBVR-Pro-SenseNova-U1</td><td><strong>0.638</strong></td><td><strong>0.811</strong></td><td><strong>0.648</strong></td><td><strong>0.695</strong></td><td><strong>0.621</strong></td><td><strong>0.770</strong></td><td><strong>0.541</strong></td><td><strong>0.464</strong></td><td><strong>0.480</strong></td><td><strong>0.328</strong></td><td><strong>0.344</strong></td><td><strong>0.558</strong></td><td><strong>0.408</strong></td></tr>
+    <tr style="border-top: 4px solid #6b7280; background-color: #e5e7eb;"><th colspan="14" align="left">Video Generation Models</th></tr>
+    <tr style="background-color: #dbeafe;"><th colspan="14" align="left">Proprietary Models</th></tr>
+    <tr><td>Veo 3.1</td><td>0.309</td><td>0.312</td><td><u>0.275</u></td><td>0.299</td><td>0.252</td><td>0.267</td><td>0.157</td><td>0.305</td><td><u>0.305</u></td><td>0.233</td><td>0.252</td><td><u>0.312</u></td><td>0.219</td></tr>
+    <tr><td>Kling V3</td><td><u>0.392</u></td><td><u>0.356</u></td><td>0.213</td><td><u>0.326</u></td><td><u>0.320</u></td><td><u>0.355</u></td><td><u>0.229</u></td><td><u>0.427</u></td><td>0.294</td><td><strong>0.564</strong></td><td><u>0.375</u></td><td>0.242</td><td><u>0.412</u></td></tr>
+    <tr><td>SeedDance 2.0</td><td><strong>0.499</strong></td><td><strong>0.451</strong></td><td><strong>0.338</strong></td><td><strong>0.361</strong></td><td><strong>0.353</strong></td><td><strong>0.468</strong></td><td><strong>0.308</strong></td><td><strong>0.547</strong></td><td><strong>0.369</strong></td><td><u>0.511</u></td><td><strong>0.478</strong></td><td><strong>0.538</strong></td><td><strong>0.532</strong></td></tr>
+    <tr style="background-color: #dbeafe;"><th colspan="14" align="left">Open-source Models</th></tr>
+    <tr><td>HunyuanVideo-I2V</td><td>0.054</td><td>0.054</td><td>0.023</td><td>0.064</td><td>0.015</td><td>0.084</td><td>0.032</td><td>0.053</td><td>0.088</td><td>0.014</td><td>0.028</td><td>0.062</td><td>0.055</td></tr>
+    <tr><td>CogVideoX1.5-5B-I2V</td><td>0.085</td><td>0.100</td><td>0.061</td><td>0.118</td><td>0.069</td><td>0.092</td><td>0.060</td><td>0.070</td><td>0.125</td><td>0.038</td><td>0.051</td><td>0.040</td><td>0.024</td></tr>
+    <tr><td>Wan2.1-I2V-14B</td><td>0.100</td><td>0.105</td><td>0.052</td><td>0.125</td><td>0.091</td><td>0.102</td><td>0.052</td><td>0.095</td><td>0.112</td><td>0.073</td><td>0.071</td><td>0.123</td><td>0.044</td></tr>
+    <tr><td>Wan2.2-TI2V-5B</td><td>0.094</td><td>0.066</td><td>0.029</td><td>0.073</td><td>0.050</td><td>0.083</td><td>0.031</td><td>0.122</td><td>0.156</td><td>0.052</td><td>0.106</td><td>0.063</td><td>0.099</td></tr>
+    <tr><td>Wan2.2-I2V-14B-720P</td><td><u>0.182</u></td><td><u>0.157</u></td><td><u>0.082</u></td><td><u>0.131</u></td><td><u>0.110</u></td><td><u>0.161</u></td><td><u>0.156</u></td><td><u>0.207</u></td><td><u>0.224</u></td><td><u>0.139</u></td><td><u>0.140</u></td><td><u>0.195</u></td><td><u>0.273</u></td></tr>
+    <tr><td>LTX2.3-I2AV</td><td>0.112</td><td>0.106</td><td>0.062</td><td>0.109</td><td>0.070</td><td>0.133</td><td>0.055</td><td>0.119</td><td>0.161</td><td>0.135</td><td>0.086</td><td>0.091</td><td>0.050</td></tr>
+    <tr><td>VBVR-Wan2.2</td><td><strong>0.517</strong></td><td><strong>0.548</strong></td><td><strong>0.237</strong></td><td><strong>0.499</strong></td><td><strong>0.334</strong></td><td><strong>0.566</strong></td><td><strong>0.591</strong></td><td><strong>0.486</strong></td><td><strong>0.310</strong></td><td><strong>0.343</strong></td><td><strong>0.345</strong></td><td><strong>0.732</strong></td><td><strong>0.684</strong></td></tr>
+    <tr style="background-color: #dbeafe;"><th colspan="14" align="left">Strong Baselines</th></tr>
+    <tr><td>VBVR-Pro-LTX2.3</td><td>0.425</td><td>0.527</td><td>0.409</td><td>0.510</td><td>0.346</td><td>0.460</td><td>0.390</td><td>0.324</td><td>0.381</td><td>0.108</td><td>0.201</td><td>0.477</td><td>0.386</td></tr>
+    <tr><td>VBVR-Pro-Wan2.1-I2V-14B</td><td><u>0.562</u></td><td><u>0.730</u></td><td><u>0.617</u></td><td><u>0.580</u></td><td><u>0.452</u></td><td><u>0.676</u></td><td><u>0.623</u></td><td><u>0.395</u></td><td><u>0.410</u></td><td><u>0.305</u></td><td><u>0.230</u></td><td><u>0.617</u></td><td><u>0.439</u></td></tr>
+    <tr><td>VBVR-Pro-Wan2.2-TI2V-5B</td><td>0.470</td><td>0.641</td><td>0.528</td><td>0.556</td><td>0.373</td><td>0.565</td><td>0.557</td><td>0.300</td><td>0.333</td><td>0.127</td><td>0.161</td><td>0.505</td><td>0.409</td></tr>
+    <tr><td>VBVR-Pro-Wan2.2-I2V-14B</td><td><strong>0.670</strong></td><td><strong>0.808</strong></td><td><strong>0.632</strong></td><td><strong>0.685</strong></td><td><strong>0.556</strong></td><td><strong>0.751</strong></td><td><strong>0.636</strong></td><td><strong>0.532</strong></td><td><strong>0.479</strong></td><td><strong>0.418</strong></td><td><strong>0.350</strong></td><td><strong>0.679</strong></td><td><strong>0.690</strong></td></tr>
+  </tbody>
+</table>
+
+## Quick Start
+
+### Method 1: Standalone Diffusers inference
+
+#### 1. Install Diffusers
+
+```bash
+pip install torch torchvision
+pip install git+https://github.com/huggingface/diffusers
+```
+
+#### 2. Run `example.py`
+
+The included [`example.py`](example.py) loads the merged checkpoint directly
+with Diffusers and enables model CPU offloading.
+
+```bash
+python example.py \
+  --model_path Video-Reason/VBVR-Pro-FLUX2-dev \
+  --image input.png \
+  --prompt "Move the red block to the left of the blue block." \
+  --output output.png
+```
+
+### Method 2: Unified VBVR-Pro inference
+
+Clone [Video-Reason/VBVR-Pro](https://github.com/Video-Reason/VBVR-Pro) and
+create its unified inference environment:
+
+```bash
+git clone https://github.com/Video-Reason/VBVR-Pro.git
+cd VBVR-Pro/
+uv sync --extra cu124 # or one of [cu118|cu121|cu124|cu126|cu128|cu129]
+source .venv/bin/activate
+```
+
+Then run the unified inference script:
+
+```bash
+python example.py \
+  --model_path Video-Reason/VBVR-Pro-FLUX2-dev \
+  --image_paths input.png \
+  --prompt "Move the red block to the left of the blue block." \
+  --output output.png
+```
+
+## Citation
+
+```bibtex
+@article{vbvr2025,
+  title={VBVR: A Very Big Video Reasoning Suite},
+  year={2025},
+}
+```

example.py

@@ -0,0 +1,39 @@
+"""VBVR-Pro-FLUX2-dev inference example.
+
+Usage:
+    python example.py --model_path ./VBVR-Pro-FLUX2-dev
+"""
+
+import argparse
+import torch
+from PIL import Image
+from diffusers import Flux2Pipeline
+
+parser = argparse.ArgumentParser()
+parser.add_argument("--model_path", type=str, default="VBVR-Pro-FLUX2-dev")
+parser.add_argument("--image", type=str, required=True, help="Path to input image")
+parser.add_argument("--prompt", type=str, required=True, help="Editing instruction")
+parser.add_argument("--output", type=str, default="output.png")
+parser.add_argument("--steps", type=int, default=50)
+parser.add_argument("--seed", type=int, default=42)
+args = parser.parse_args()
+
+print(f"Loading model from: {args.model_path}")
+pipe = Flux2Pipeline.from_pretrained(
+    args.model_path, torch_dtype=torch.bfloat16
+)
+pipe.enable_model_cpu_offload()
+
+image = Image.open(args.image).convert("RGB")
+print(f"Input image: {args.image} ({image.size[0]}x{image.size[1]})")
+
+output = pipe(
+    image=image,
+    prompt=args.prompt,
+    num_inference_steps=args.steps,
+    guidance_scale=2.5,
+    generator=torch.manual_seed(args.seed),
+).images[0]
+
+output.save(args.output)
+print(f"Saved to: {args.output}")

model_index.json

@@ -0,0 +1,24 @@
+{
+  "_class_name": "Flux2Pipeline",
+  "_diffusers_version": "0.36.0.dev0",
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "text_encoder": [
+    "transformers",
+    "Mistral3ForConditionalGeneration"
+  ],
+  "tokenizer": [
+    "transformers",
+    "PixtralProcessor"
+  ],
+  "transformer": [
+    "diffusers",
+    "Flux2Transformer2DModel"
+  ],
+  "vae": [
+    "diffusers",
+    "AutoencoderKLFlux2"
+  ]
+}

scheduler/scheduler_config.json

@@ -0,0 +1,18 @@
+{
+  "_class_name": "FlowMatchEulerDiscreteScheduler",
+  "_diffusers_version": "0.36.0.dev0",
+  "base_image_seq_len": 256,
+  "base_shift": 0.5,
+  "invert_sigmas": false,
+  "max_image_seq_len": 4096,
+  "max_shift": 1.15,
+  "num_train_timesteps": 1000,
+  "shift": 3.0,
+  "shift_terminal": null,
+  "stochastic_sampling": false,
+  "time_shift_type": "exponential",
+  "use_beta_sigmas": false,
+  "use_dynamic_shifting": true,
+  "use_exponential_sigmas": false,
+  "use_karras_sigmas": false
+}

text_encoder/config.json

@@ -0,0 +1,48 @@
+{
+  "architectures": [
+    "Mistral3ForConditionalGeneration"
+  ],
+  "dtype": "bfloat16",
+  "image_token_index": 10,
+  "model_type": "mistral3",
+  "multimodal_projector_bias": false,
+  "projector_hidden_act": "gelu",
+  "spatial_merge_size": 2,
+  "text_config": {
+    "attention_dropout": 0.0,
+    "dtype": "bfloat16",
+    "head_dim": 128,
+    "hidden_act": "silu",
+    "hidden_size": 5120,
+    "initializer_range": 0.02,
+    "intermediate_size": 32768,
+    "max_position_embeddings": 131072,
+    "model_type": "mistral",
+    "num_attention_heads": 32,
+    "num_hidden_layers": 40,
+    "num_key_value_heads": 8,
+    "rms_norm_eps": 1e-05,
+    "rope_theta": 1000000000.0,
+    "sliding_window": null,
+    "use_cache": true,
+    "vocab_size": 131072
+  },
+  "transformers_version": "4.57.1",
+  "vision_config": {
+    "attention_dropout": 0.0,
+    "dtype": "bfloat16",
+    "head_dim": 64,
+    "hidden_act": "silu",
+    "hidden_size": 1024,
+    "image_size": 1540,
+    "initializer_range": 0.02,
+    "intermediate_size": 4096,
+    "model_type": "pixtral",
+    "num_attention_heads": 16,
+    "num_channels": 3,
+    "num_hidden_layers": 24,
+    "patch_size": 14,
+    "rope_theta": 10000.0
+  },
+  "vision_feature_layer": -1
+}

text_encoder/generation_config.json

@@ -0,0 +1,8 @@
+{
+  "_from_model_config": true,
+  "bos_token_id": 1,
+  "do_sample": true,
+  "eos_token_id": 2,
+  "temperature": 0.15,
+  "transformers_version": "4.57.1"
+}

text_encoder/model-00001-of-00010.safetensors

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

text_encoder/model-00002-of-00010.safetensors

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

text_encoder/model-00003-of-00010.safetensors

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

text_encoder/model-00004-of-00010.safetensors

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

text_encoder/model-00005-of-00010.safetensors

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

text_encoder/model-00006-of-00010.safetensors

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

text_encoder/model-00007-of-00010.safetensors

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

text_encoder/model-00008-of-00010.safetensors

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

text_encoder/model-00009-of-00010.safetensors

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

text_encoder/model-00010-of-00010.safetensors

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

text_encoder/model.safetensors.index.json

@@ -0,0 +1,593 @@
+{
+  "metadata": {
+    "total_parameters": 24011361280,
+    "total_size": 48022722560
+  },
+  "weight_map": {
+    "language_model.lm_head.weight": "model-00010-of-00010.safetensors",
+    "language_model.model.embed_tokens.weight": "model-00001-of-00010.safetensors",
+    "language_model.model.layers.0.input_layernorm.weight": "model-00001-of-00010.safetensors",
+    "language_model.model.layers.0.mlp.down_proj.weight": "model-00001-of-00010.safetensors",
+    "language_model.model.layers.0.mlp.gate_proj.weight": "model-00001-of-00010.safetensors",
+    "language_model.model.layers.0.mlp.up_proj.weight": "model-00001-of-00010.safetensors",
+    "language_model.model.layers.0.post_attention_layernorm.weight": "model-00001-of-00010.safetensors",
+    "language_model.model.layers.0.self_attn.k_proj.weight": "model-00001-of-00010.safetensors",
+    "language_model.model.layers.0.self_attn.o_proj.weight": "model-00001-of-00010.safetensors",
+    "language_model.model.layers.0.self_attn.q_proj.weight": "model-00001-of-00010.safetensors",
+    "language_model.model.layers.0.self_attn.v_proj.weight": "model-00001-of-00010.safetensors",
+    "language_model.model.layers.1.input_layernorm.weight": "model-00001-of-00010.safetensors",
+    "language_model.model.layers.1.mlp.down_proj.weight": "model-00001-of-00010.safetensors",
+    "language_model.model.layers.1.mlp.gate_proj.weight": "model-00001-of-00010.safetensors",
+    "language_model.model.layers.1.mlp.up_proj.weight": "model-00001-of-00010.safetensors",
+    "language_model.model.layers.1.post_attention_layernorm.weight": "model-00001-of-00010.safetensors",
+    "language_model.model.layers.1.self_attn.k_proj.weight": "model-00001-of-00010.safetensors",
+    "language_model.model.layers.1.self_attn.o_proj.weight": "model-00001-of-00010.safetensors",
+    "language_model.model.layers.1.self_attn.q_proj.weight": "model-00001-of-00010.safetensors",
+    "language_model.model.layers.1.self_attn.v_proj.weight": "model-00001-of-00010.safetensors",
+    "language_model.model.layers.10.input_layernorm.weight": "model-00003-of-00010.safetensors",
+    "language_model.model.layers.10.mlp.down_proj.weight": "model-00003-of-00010.safetensors",
+    "language_model.model.layers.10.mlp.gate_proj.weight": "model-00003-of-00010.safetensors",
+    "language_model.model.layers.10.mlp.up_proj.weight": "model-00003-of-00010.safetensors",
+    "language_model.model.layers.10.post_attention_layernorm.weight": "model-00003-of-00010.safetensors",
+    "language_model.model.layers.10.self_attn.k_proj.weight": "model-00003-of-00010.safetensors",
+    "language_model.model.layers.10.self_attn.o_proj.weight": "model-00003-of-00010.safetensors",
+    "language_model.model.layers.10.self_attn.q_proj.weight": "model-00003-of-00010.safetensors",
+    "language_model.model.layers.10.self_attn.v_proj.weight": "model-00003-of-00010.safetensors",
+    "language_model.model.layers.11.input_layernorm.weight": "model-00004-of-00010.safetensors",
+    "language_model.model.layers.11.mlp.down_proj.weight": "model-00004-of-00010.safetensors",
+    "language_model.model.layers.11.mlp.gate_proj.weight": "model-00004-of-00010.safetensors",
+    "language_model.model.layers.11.mlp.up_proj.weight": "model-00004-of-00010.safetensors",
+    "language_model.model.layers.11.post_attention_layernorm.weight": "model-00004-of-00010.safetensors",
+    "language_model.model.layers.11.self_attn.k_proj.weight": "model-00003-of-00010.safetensors",
+    "language_model.model.layers.11.self_attn.o_proj.weight": "model-00003-of-00010.safetensors",
+    "language_model.model.layers.11.self_attn.q_proj.weight": "model-00003-of-00010.safetensors",
+    "language_model.model.layers.11.self_attn.v_proj.weight": "model-00003-of-00010.safetensors",
+    "language_model.model.layers.12.input_layernorm.weight": "model-00004-of-00010.safetensors",
+    "language_model.model.layers.12.mlp.down_proj.weight": "model-00004-of-00010.safetensors",
+    "language_model.model.layers.12.mlp.gate_proj.weight": "model-00004-of-00010.safetensors",
+    "language_model.model.layers.12.mlp.up_proj.weight": "model-00004-of-00010.safetensors",
+    "language_model.model.layers.12.post_attention_layernorm.weight": "model-00004-of-00010.safetensors",
+    "language_model.model.layers.12.self_attn.k_proj.weight": "model-00004-of-00010.safetensors",
+    "language_model.model.layers.12.self_attn.o_proj.weight": "model-00004-of-00010.safetensors",
+    "language_model.model.layers.12.self_attn.q_proj.weight": "model-00004-of-00010.safetensors",
+    "language_model.model.layers.12.self_attn.v_proj.weight": "model-00004-of-00010.safetensors",
+    "language_model.model.layers.13.input_layernorm.weight": "model-00004-of-00010.safetensors",
+    "language_model.model.layers.13.mlp.down_proj.weight": "model-00004-of-00010.safetensors",
+    "language_model.model.layers.13.mlp.gate_proj.weight": "model-00004-of-00010.safetensors",
+    "language_model.model.layers.13.mlp.up_proj.weight": "model-00004-of-00010.safetensors",
+    "language_model.model.layers.13.post_attention_layernorm.weight": "model-00004-of-00010.safetensors",
+    "language_model.model.layers.13.self_attn.k_proj.weight": "model-00004-of-00010.safetensors",
+    "language_model.model.layers.13.self_attn.o_proj.weight": "model-00004-of-00010.safetensors",
+    "language_model.model.layers.13.self_attn.q_proj.weight": "model-00004-of-00010.safetensors",
+    "language_model.model.layers.13.self_attn.v_proj.weight": "model-00004-of-00010.safetensors",
+    "language_model.model.layers.14.input_layernorm.weight": "model-00004-of-00010.safetensors",
+    "language_model.model.layers.14.mlp.down_proj.weight": "model-00004-of-00010.safetensors",
+    "language_model.model.layers.14.mlp.gate_proj.weight": "model-00004-of-00010.safetensors",
+    "language_model.model.layers.14.mlp.up_proj.weight": "model-00004-of-00010.safetensors",
+    "language_model.model.layers.14.post_attention_layernorm.weight": "model-00004-of-00010.safetensors",
+    "language_model.model.layers.14.self_attn.k_proj.weight": "model-00004-of-00010.safetensors",
+    "language_model.model.layers.14.self_attn.o_proj.weight": "model-00004-of-00010.safetensors",
+    "language_model.model.layers.14.self_attn.q_proj.weight": "model-00004-of-00010.safetensors",
+    "language_model.model.layers.14.self_attn.v_proj.weight": "model-00004-of-00010.safetensors",
+    "language_model.model.layers.15.input_layernorm.weight": "model-00005-of-00010.safetensors",
+    "language_model.model.layers.15.mlp.down_proj.weight": "model-00005-of-00010.safetensors",
+    "language_model.model.layers.15.mlp.gate_proj.weight": "model-00004-of-00010.safetensors",
+    "language_model.model.layers.15.mlp.up_proj.weight": "model-00005-of-00010.safetensors",
+    "language_model.model.layers.15.post_attention_layernorm.weight": "model-00005-of-00010.safetensors",
+    "language_model.model.layers.15.self_attn.k_proj.weight": "model-00004-of-00010.safetensors",
+    "language_model.model.layers.15.self_attn.o_proj.weight": "model-00004-of-00010.safetensors",
+    "language_model.model.layers.15.self_attn.q_proj.weight": "model-00004-of-00010.safetensors",
+    "language_model.model.layers.15.self_attn.v_proj.weight": "model-00004-of-00010.safetensors",
+    "language_model.model.layers.16.input_layernorm.weight": "model-00005-of-00010.safetensors",
+    "language_model.model.layers.16.mlp.down_proj.weight": "model-00005-of-00010.safetensors",
+    "language_model.model.layers.16.mlp.gate_proj.weight": "model-00005-of-00010.safetensors",
+    "language_model.model.layers.16.mlp.up_proj.weight": "model-00005-of-00010.safetensors",
+    "language_model.model.layers.16.post_attention_layernorm.weight": "model-00005-of-00010.safetensors",
+    "language_model.model.layers.16.self_attn.k_proj.weight": "model-00005-of-00010.safetensors",
+    "language_model.model.layers.16.self_attn.o_proj.weight": "model-00005-of-00010.safetensors",
+    "language_model.model.layers.16.self_attn.q_proj.weight": "model-00005-of-00010.safetensors",
+    "language_model.model.layers.16.self_attn.v_proj.weight": "model-00005-of-00010.safetensors",
+    "language_model.model.layers.17.input_layernorm.weight": "model-00005-of-00010.safetensors",
+    "language_model.model.layers.17.mlp.down_proj.weight": "model-00005-of-00010.safetensors",
+    "language_model.model.layers.17.mlp.gate_proj.weight": "model-00005-of-00010.safetensors",
+    "language_model.model.layers.17.mlp.up_proj.weight": "model-00005-of-00010.safetensors",
+    "language_model.model.layers.17.post_attention_layernorm.weight": "model-00005-of-00010.safetensors",
+    "language_model.model.layers.17.self_attn.k_proj.weight": "model-00005-of-00010.safetensors",
+    "language_model.model.layers.17.self_attn.o_proj.weight": "model-00005-of-00010.safetensors",
+    "language_model.model.layers.17.self_attn.q_proj.weight": "model-00005-of-00010.safetensors",
+    "language_model.model.layers.17.self_attn.v_proj.weight": "model-00005-of-00010.safetensors",
+    "language_model.model.layers.18.input_layernorm.weight": "model-00005-of-00010.safetensors",
+    "language_model.model.layers.18.mlp.down_proj.weight": "model-00005-of-00010.safetensors",
+    "language_model.model.layers.18.mlp.gate_proj.weight": "model-00005-of-00010.safetensors",
+    "language_model.model.layers.18.mlp.up_proj.weight": "model-00005-of-00010.safetensors",
+    "language_model.model.layers.18.post_attention_layernorm.weight": "model-00005-of-00010.safetensors",
+    "language_model.model.layers.18.self_attn.k_proj.weight": "model-00005-of-00010.safetensors",
+    "language_model.model.layers.18.self_attn.o_proj.weight": "model-00005-of-00010.safetensors",
+    "language_model.model.layers.18.self_attn.q_proj.weight": "model-00005-of-00010.safetensors",
+    "language_model.model.layers.18.self_attn.v_proj.weight": "model-00005-of-00010.safetensors",
+    "language_model.model.layers.19.input_layernorm.weight": "model-00006-of-00010.safetensors",
+    "language_model.model.layers.19.mlp.down_proj.weight": "model-00006-of-00010.safetensors",
+    "language_model.model.layers.19.mlp.gate_proj.weight": "model-00005-of-00010.safetensors",
+    "language_model.model.layers.19.mlp.up_proj.weight": "model-00005-of-00010.safetensors",
+    "language_model.model.layers.19.post_attention_layernorm.weight": "model-00006-of-00010.safetensors",
+    "language_model.model.layers.19.self_attn.k_proj.weight": "model-00005-of-00010.safetensors",
+    "language_model.model.layers.19.self_attn.o_proj.weight": "model-00005-of-00010.safetensors",
+    "language_model.model.layers.19.self_attn.q_proj.weight": "model-00005-of-00010.safetensors",
+    "language_model.model.layers.19.self_attn.v_proj.weight": "model-00005-of-00010.safetensors",
+    "language_model.model.layers.2.input_layernorm.weight": "model-00002-of-00010.safetensors",
+    "language_model.model.layers.2.mlp.down_proj.weight": "model-00002-of-00010.safetensors",
+    "language_model.model.layers.2.mlp.gate_proj.weight": "model-00001-of-00010.safetensors",
+    "language_model.model.layers.2.mlp.up_proj.weight": "model-00002-of-00010.safetensors",
+    "language_model.model.layers.2.post_attention_layernorm.weight": "model-00002-of-00010.safetensors",
+    "language_model.model.layers.2.self_attn.k_proj.weight": "model-00001-of-00010.safetensors",
+    "language_model.model.layers.2.self_attn.o_proj.weight": "model-00001-of-00010.safetensors",
+    "language_model.model.layers.2.self_attn.q_proj.weight": "model-00001-of-00010.safetensors",
+    "language_model.model.layers.2.self_attn.v_proj.weight": "model-00001-of-00010.safetensors",
+    "language_model.model.layers.20.input_layernorm.weight": "model-00006-of-00010.safetensors",
+    "language_model.model.layers.20.mlp.down_proj.weight": "model-00006-of-00010.safetensors",
+    "language_model.model.layers.20.mlp.gate_proj.weight": "model-00006-of-00010.safetensors",
+    "language_model.model.layers.20.mlp.up_proj.weight": "model-00006-of-00010.safetensors",
+    "language_model.model.layers.20.post_attention_layernorm.weight": "model-00006-of-00010.safetensors",
+    "language_model.model.layers.20.self_attn.k_proj.weight": "model-00006-of-00010.safetensors",
+    "language_model.model.layers.20.self_attn.o_proj.weight": "model-00006-of-00010.safetensors",
+    "language_model.model.layers.20.self_attn.q_proj.weight": "model-00006-of-00010.safetensors",
+    "language_model.model.layers.20.self_attn.v_proj.weight": "model-00006-of-00010.safetensors",
+    "language_model.model.layers.21.input_layernorm.weight": "model-00006-of-00010.safetensors",
+    "language_model.model.layers.21.mlp.down_proj.weight": "model-00006-of-00010.safetensors",
+    "language_model.model.layers.21.mlp.gate_proj.weight": "model-00006-of-00010.safetensors",
+    "language_model.model.layers.21.mlp.up_proj.weight": "model-00006-of-00010.safetensors",
+    "language_model.model.layers.21.post_attention_layernorm.weight": "model-00006-of-00010.safetensors",
+    "language_model.model.layers.21.self_attn.k_proj.weight": "model-00006-of-00010.safetensors",
+    "language_model.model.layers.21.self_attn.o_proj.weight": "model-00006-of-00010.safetensors",
+    "language_model.model.layers.21.self_attn.q_proj.weight": "model-00006-of-00010.safetensors",
+    "language_model.model.layers.21.self_attn.v_proj.weight": "model-00006-of-00010.safetensors",
+    "language_model.model.layers.22.input_layernorm.weight": "model-00006-of-00010.safetensors",
+    "language_model.model.layers.22.mlp.down_proj.weight": "model-00006-of-00010.safetensors",
+    "language_model.model.layers.22.mlp.gate_proj.weight": "model-00006-of-00010.safetensors",
+    "language_model.model.layers.22.mlp.up_proj.weight": "model-00006-of-00010.safetensors",
+    "language_model.model.layers.22.post_attention_layernorm.weight": "model-00006-of-00010.safetensors",
+    "language_model.model.layers.22.self_attn.k_proj.weight": "model-00006-of-00010.safetensors",
+    "language_model.model.layers.22.self_attn.o_proj.weight": "model-00006-of-00010.safetensors",
+    "language_model.model.layers.22.self_attn.q_proj.weight": "model-00006-of-00010.safetensors",
+    "language_model.model.layers.22.self_attn.v_proj.weight": "model-00006-of-00010.safetensors",
+    "language_model.model.layers.23.input_layernorm.weight": "model-00006-of-00010.safetensors",
+    "language_model.model.layers.23.mlp.down_proj.weight": "model-00006-of-00010.safetensors",
+    "language_model.model.layers.23.mlp.gate_proj.weight": "model-00006-of-00010.safetensors",
+    "language_model.model.layers.23.mlp.up_proj.weight": "model-00006-of-00010.safetensors",
+    "language_model.model.layers.23.post_attention_layernorm.weight": "model-00006-of-00010.safetensors",
+    "language_model.model.layers.23.self_attn.k_proj.weight": "model-00006-of-00010.safetensors",
+    "language_model.model.layers.23.self_attn.o_proj.weight": "model-00006-of-00010.safetensors",
+    "language_model.model.layers.23.self_attn.q_proj.weight": "model-00006-of-00010.safetensors",
+    "language_model.model.layers.23.self_attn.v_proj.weight": "model-00006-of-00010.safetensors",
+    "language_model.model.layers.24.input_layernorm.weight": "model-00007-of-00010.safetensors",
+    "language_model.model.layers.24.mlp.down_proj.weight": "model-00007-of-00010.safetensors",
+    "language_model.model.layers.24.mlp.gate_proj.weight": "model-00007-of-00010.safetensors",
+    "language_model.model.layers.24.mlp.up_proj.weight": "model-00007-of-00010.safetensors",
+    "language_model.model.layers.24.post_attention_layernorm.weight": "model-00007-of-00010.safetensors",
+    "language_model.model.layers.24.self_attn.k_proj.weight": "model-00006-of-00010.safetensors",
+    "language_model.model.layers.24.self_attn.o_proj.weight": "model-00006-of-00010.safetensors",
+    "language_model.model.layers.24.self_attn.q_proj.weight": "model-00006-of-00010.safetensors",
+    "language_model.model.layers.24.self_attn.v_proj.weight": "model-00006-of-00010.safetensors",
+    "language_model.model.layers.25.input_layernorm.weight": "model-00007-of-00010.safetensors",
+    "language_model.model.layers.25.mlp.down_proj.weight": "model-00007-of-00010.safetensors",
+    "language_model.model.layers.25.mlp.gate_proj.weight": "model-00007-of-00010.safetensors",
+    "language_model.model.layers.25.mlp.up_proj.weight": "model-00007-of-00010.safetensors",
+    "language_model.model.layers.25.post_attention_layernorm.weight": "model-00007-of-00010.safetensors",
+    "language_model.model.layers.25.self_attn.k_proj.weight": "model-00007-of-00010.safetensors",
+    "language_model.model.layers.25.self_attn.o_proj.weight": "model-00007-of-00010.safetensors",
+    "language_model.model.layers.25.self_attn.q_proj.weight": "model-00007-of-00010.safetensors",
+    "language_model.model.layers.25.self_attn.v_proj.weight": "model-00007-of-00010.safetensors",
+    "language_model.model.layers.26.input_layernorm.weight": "model-00007-of-00010.safetensors",
+    "language_model.model.layers.26.mlp.down_proj.weight": "model-00007-of-00010.safetensors",
+    "language_model.model.layers.26.mlp.gate_proj.weight": "model-00007-of-00010.safetensors",
+    "language_model.model.layers.26.mlp.up_proj.weight": "model-00007-of-00010.safetensors",
+    "language_model.model.layers.26.post_attention_layernorm.weight": "model-00007-of-00010.safetensors",
+    "language_model.model.layers.26.self_attn.k_proj.weight": "model-00007-of-00010.safetensors",
+    "language_model.model.layers.26.self_attn.o_proj.weight": "model-00007-of-00010.safetensors",
+    "language_model.model.layers.26.self_attn.q_proj.weight": "model-00007-of-00010.safetensors",
+    "language_model.model.layers.26.self_attn.v_proj.weight": "model-00007-of-00010.safetensors",
+    "language_model.model.layers.27.input_layernorm.weight": "model-00007-of-00010.safetensors",
+    "language_model.model.layers.27.mlp.down_proj.weight": "model-00007-of-00010.safetensors",
+    "language_model.model.layers.27.mlp.gate_proj.weight": "model-00007-of-00010.safetensors",
+    "language_model.model.layers.27.mlp.up_proj.weight": "model-00007-of-00010.safetensors",
+    "language_model.model.layers.27.post_attention_layernorm.weight": "model-00007-of-00010.safetensors",
+    "language_model.model.layers.27.self_attn.k_proj.weight": "model-00007-of-00010.safetensors",
+    "language_model.model.layers.27.self_attn.o_proj.weight": "model-00007-of-00010.safetensors",
+    "language_model.model.layers.27.self_attn.q_proj.weight": "model-00007-of-00010.safetensors",
+    "language_model.model.layers.27.self_attn.v_proj.weight": "model-00007-of-00010.safetensors",
+    "language_model.model.layers.28.input_layernorm.weight": "model-00008-of-00010.safetensors",
+    "language_model.model.layers.28.mlp.down_proj.weight": "model-00008-of-00010.safetensors",
+    "language_model.model.layers.28.mlp.gate_proj.weight": "model-00007-of-00010.safetensors",
+    "language_model.model.layers.28.mlp.up_proj.weight": "model-00008-of-00010.safetensors",
+    "language_model.model.layers.28.post_attention_layernorm.weight": "model-00008-of-00010.safetensors",
+    "language_model.model.layers.28.self_attn.k_proj.weight": "model-00007-of-00010.safetensors",
+    "language_model.model.layers.28.self_attn.o_proj.weight": "model-00007-of-00010.safetensors",
+    "language_model.model.layers.28.self_attn.q_proj.weight": "model-00007-of-00010.safetensors",
+    "language_model.model.layers.28.self_attn.v_proj.weight": "model-00007-of-00010.safetensors",
+    "language_model.model.layers.29.input_layernorm.weight": "model-00008-of-00010.safetensors",
+    "language_model.model.layers.29.mlp.down_proj.weight": "model-00008-of-00010.safetensors",
+    "language_model.model.layers.29.mlp.gate_proj.weight": "model-00008-of-00010.safetensors",
+    "language_model.model.layers.29.mlp.up_proj.weight": "model-00008-of-00010.safetensors",
+    "language_model.model.layers.29.post_attention_layernorm.weight": "model-00008-of-00010.safetensors",
+    "language_model.model.layers.29.self_attn.k_proj.weight": "model-00008-of-00010.safetensors",
+    "language_model.model.layers.29.self_attn.o_proj.weight": "model-00008-of-00010.safetensors",
+    "language_model.model.layers.29.self_attn.q_proj.weight": "model-00008-of-00010.safetensors",
+    "language_model.model.layers.29.self_attn.v_proj.weight": "model-00008-of-00010.safetensors",
+    "language_model.model.layers.3.input_layernorm.weight": "model-00002-of-00010.safetensors",
+    "language_model.model.layers.3.mlp.down_proj.weight": "model-00002-of-00010.safetensors",
+    "language_model.model.layers.3.mlp.gate_proj.weight": "model-00002-of-00010.safetensors",
+    "language_model.model.layers.3.mlp.up_proj.weight": "model-00002-of-00010.safetensors",
+    "language_model.model.layers.3.post_attention_layernorm.weight": "model-00002-of-00010.safetensors",
+    "language_model.model.layers.3.self_attn.k_proj.weight": "model-00002-of-00010.safetensors",
+    "language_model.model.layers.3.self_attn.o_proj.weight": "model-00002-of-00010.safetensors",
+    "language_model.model.layers.3.self_attn.q_proj.weight": "model-00002-of-00010.safetensors",
+    "language_model.model.layers.3.self_attn.v_proj.weight": "model-00002-of-00010.safetensors",
+    "language_model.model.layers.30.input_layernorm.weight": "model-00008-of-00010.safetensors",
+    "language_model.model.layers.30.mlp.down_proj.weight": "model-00008-of-00010.safetensors",
+    "language_model.model.layers.30.mlp.gate_proj.weight": "model-00008-of-00010.safetensors",
+    "language_model.model.layers.30.mlp.up_proj.weight": "model-00008-of-00010.safetensors",
+    "language_model.model.layers.30.post_attention_layernorm.weight": "model-00008-of-00010.safetensors",
+    "language_model.model.layers.30.self_attn.k_proj.weight": "model-00008-of-00010.safetensors",
+    "language_model.model.layers.30.self_attn.o_proj.weight": "model-00008-of-00010.safetensors",
+    "language_model.model.layers.30.self_attn.q_proj.weight": "model-00008-of-00010.safetensors",
+    "language_model.model.layers.30.self_attn.v_proj.weight": "model-00008-of-00010.safetensors",
+    "language_model.model.layers.31.input_layernorm.weight": "model-00008-of-00010.safetensors",
+    "language_model.model.layers.31.mlp.down_proj.weight": "model-00008-of-00010.safetensors",
+    "language_model.model.layers.31.mlp.gate_proj.weight": "model-00008-of-00010.safetensors",
+    "language_model.model.layers.31.mlp.up_proj.weight": "model-00008-of-00010.safetensors",
+    "language_model.model.layers.31.post_attention_layernorm.weight": "model-00008-of-00010.safetensors",
+    "language_model.model.layers.31.self_attn.k_proj.weight": "model-00008-of-00010.safetensors",
+    "language_model.model.layers.31.self_attn.o_proj.weight": "model-00008-of-00010.safetensors",
+    "language_model.model.layers.31.self_attn.q_proj.weight": "model-00008-of-00010.safetensors",
+    "language_model.model.layers.31.self_attn.v_proj.weight": "model-00008-of-00010.safetensors",
+    "language_model.model.layers.32.input_layernorm.weight": "model-00009-of-00010.safetensors",
+    "language_model.model.layers.32.mlp.down_proj.weight": "model-00009-of-00010.safetensors",
+    "language_model.model.layers.32.mlp.gate_proj.weight": "model-00008-of-00010.safetensors",
+    "language_model.model.layers.32.mlp.up_proj.weight": "model-00008-of-00010.safetensors",
+    "language_model.model.layers.32.post_attention_layernorm.weight": "model-00009-of-00010.safetensors",
+    "language_model.model.layers.32.self_attn.k_proj.weight": "model-00008-of-00010.safetensors",
+    "language_model.model.layers.32.self_attn.o_proj.weight": "model-00008-of-00010.safetensors",
+    "language_model.model.layers.32.self_attn.q_proj.weight": "model-00008-of-00010.safetensors",
+    "language_model.model.layers.32.self_attn.v_proj.weight": "model-00008-of-00010.safetensors",
+    "language_model.model.layers.33.input_layernorm.weight": "model-00009-of-00010.safetensors",
+    "language_model.model.layers.33.mlp.down_proj.weight": "model-00009-of-00010.safetensors",
+    "language_model.model.layers.33.mlp.gate_proj.weight": "model-00009-of-00010.safetensors",
+    "language_model.model.layers.33.mlp.up_proj.weight": "model-00009-of-00010.safetensors",
+    "language_model.model.layers.33.post_attention_layernorm.weight": "model-00009-of-00010.safetensors",
+    "language_model.model.layers.33.self_attn.k_proj.weight": "model-00009-of-00010.safetensors",
+    "language_model.model.layers.33.self_attn.o_proj.weight": "model-00009-of-00010.safetensors",
+    "language_model.model.layers.33.self_attn.q_proj.weight": "model-00009-of-00010.safetensors",
+    "language_model.model.layers.33.self_attn.v_proj.weight": "model-00009-of-00010.safetensors",
+    "language_model.model.layers.34.input_layernorm.weight": "model-00009-of-00010.safetensors",
+    "language_model.model.layers.34.mlp.down_proj.weight": "model-00009-of-00010.safetensors",
+    "language_model.model.layers.34.mlp.gate_proj.weight": "model-00009-of-00010.safetensors",
+    "language_model.model.layers.34.mlp.up_proj.weight": "model-00009-of-00010.safetensors",
+    "language_model.model.layers.34.post_attention_layernorm.weight": "model-00009-of-00010.safetensors",
+    "language_model.model.layers.34.self_attn.k_proj.weight": "model-00009-of-00010.safetensors",
+    "language_model.model.layers.34.self_attn.o_proj.weight": "model-00009-of-00010.safetensors",
+    "language_model.model.layers.34.self_attn.q_proj.weight": "model-00009-of-00010.safetensors",
+    "language_model.model.layers.34.self_attn.v_proj.weight": "model-00009-of-00010.safetensors",
+    "language_model.model.layers.35.input_layernorm.weight": "model-00009-of-00010.safetensors",
+    "language_model.model.layers.35.mlp.down_proj.weight": "model-00009-of-00010.safetensors",
+    "language_model.model.layers.35.mlp.gate_proj.weight": "model-00009-of-00010.safetensors",
+    "language_model.model.layers.35.mlp.up_proj.weight": "model-00009-of-00010.safetensors",
+    "language_model.model.layers.35.post_attention_layernorm.weight": "model-00009-of-00010.safetensors",
+    "language_model.model.layers.35.self_attn.k_proj.weight": "model-00009-of-00010.safetensors",
+    "language_model.model.layers.35.self_attn.o_proj.weight": "model-00009-of-00010.safetensors",
+    "language_model.model.layers.35.self_attn.q_proj.weight": "model-00009-of-00010.safetensors",
+    "language_model.model.layers.35.self_attn.v_proj.weight": "model-00009-of-00010.safetensors",
+    "language_model.model.layers.36.input_layernorm.weight": "model-00009-of-00010.safetensors",
+    "language_model.model.layers.36.mlp.down_proj.weight": "model-00009-of-00010.safetensors",
+    "language_model.model.layers.36.mlp.gate_proj.weight": "model-00009-of-00010.safetensors",
+    "language_model.model.layers.36.mlp.up_proj.weight": "model-00009-of-00010.safetensors",
+    "language_model.model.layers.36.post_attention_layernorm.weight": "model-00009-of-00010.safetensors",
+    "language_model.model.layers.36.self_attn.k_proj.weight": "model-00009-of-00010.safetensors",
+    "language_model.model.layers.36.self_attn.o_proj.weight": "model-00009-of-00010.safetensors",
+    "language_model.model.layers.36.self_attn.q_proj.weight": "model-00009-of-00010.safetensors",
+    "language_model.model.layers.36.self_attn.v_proj.weight": "model-00009-of-00010.safetensors",
+    "language_model.model.layers.37.input_layernorm.weight": "model-00010-of-00010.safetensors",
+    "language_model.model.layers.37.mlp.down_proj.weight": "model-00010-of-00010.safetensors",
+    "language_model.model.layers.37.mlp.gate_proj.weight": "model-00010-of-00010.safetensors",
+    "language_model.model.layers.37.mlp.up_proj.weight": "model-00010-of-00010.safetensors",
+    "language_model.model.layers.37.post_attention_layernorm.weight": "model-00010-of-00010.safetensors",
+    "language_model.model.layers.37.self_attn.k_proj.weight": "model-00009-of-00010.safetensors",
+    "language_model.model.layers.37.self_attn.o_proj.weight": "model-00009-of-00010.safetensors",
+    "language_model.model.layers.37.self_attn.q_proj.weight": "model-00009-of-00010.safetensors",
+    "language_model.model.layers.37.self_attn.v_proj.weight": "model-00009-of-00010.safetensors",
+    "language_model.model.layers.38.input_layernorm.weight": "model-00010-of-00010.safetensors",
+    "language_model.model.layers.38.mlp.down_proj.weight": "model-00010-of-00010.safetensors",
+    "language_model.model.layers.38.mlp.gate_proj.weight": "model-00010-of-00010.safetensors",
+    "language_model.model.layers.38.mlp.up_proj.weight": "model-00010-of-00010.safetensors",
+    "language_model.model.layers.38.post_attention_layernorm.weight": "model-00010-of-00010.safetensors",
+    "language_model.model.layers.38.self_attn.k_proj.weight": "model-00010-of-00010.safetensors",
+    "language_model.model.layers.38.self_attn.o_proj.weight": "model-00010-of-00010.safetensors",
+    "language_model.model.layers.38.self_attn.q_proj.weight": "model-00010-of-00010.safetensors",
+    "language_model.model.layers.38.self_attn.v_proj.weight": "model-00010-of-00010.safetensors",
+    "language_model.model.layers.39.input_layernorm.weight": "model-00010-of-00010.safetensors",
+    "language_model.model.layers.39.mlp.down_proj.weight": "model-00010-of-00010.safetensors",
+    "language_model.model.layers.39.mlp.gate_proj.weight": "model-00010-of-00010.safetensors",
+    "language_model.model.layers.39.mlp.up_proj.weight": "model-00010-of-00010.safetensors",
+    "language_model.model.layers.39.post_attention_layernorm.weight": "model-00010-of-00010.safetensors",
+    "language_model.model.layers.39.self_attn.k_proj.weight": "model-00010-of-00010.safetensors",
+    "language_model.model.layers.39.self_attn.o_proj.weight": "model-00010-of-00010.safetensors",
+    "language_model.model.layers.39.self_attn.q_proj.weight": "model-00010-of-00010.safetensors",
+    "language_model.model.layers.39.self_attn.v_proj.weight": "model-00010-of-00010.safetensors",
+    "language_model.model.layers.4.input_layernorm.weight": "model-00002-of-00010.safetensors",
+    "language_model.model.layers.4.mlp.down_proj.weight": "model-00002-of-00010.safetensors",
+    "language_model.model.layers.4.mlp.gate_proj.weight": "model-00002-of-00010.safetensors",
+    "language_model.model.layers.4.mlp.up_proj.weight": "model-00002-of-00010.safetensors",
+    "language_model.model.layers.4.post_attention_layernorm.weight": "model-00002-of-00010.safetensors",
+    "language_model.model.layers.4.self_attn.k_proj.weight": "model-00002-of-00010.safetensors",
+    "language_model.model.layers.4.self_attn.o_proj.weight": "model-00002-of-00010.safetensors",
+    "language_model.model.layers.4.self_attn.q_proj.weight": "model-00002-of-00010.safetensors",
+    "language_model.model.layers.4.self_attn.v_proj.weight": "model-00002-of-00010.safetensors",
+    "language_model.model.layers.5.input_layernorm.weight": "model-00002-of-00010.safetensors",
+    "language_model.model.layers.5.mlp.down_proj.weight": "model-00002-of-00010.safetensors",
+    "language_model.model.layers.5.mlp.gate_proj.weight": "model-00002-of-00010.safetensors",
+    "language_model.model.layers.5.mlp.up_proj.weight": "model-00002-of-00010.safetensors",
+    "language_model.model.layers.5.post_attention_layernorm.weight": "model-00002-of-00010.safetensors",
+    "language_model.model.layers.5.self_attn.k_proj.weight": "model-00002-of-00010.safetensors",
+    "language_model.model.layers.5.self_attn.o_proj.weight": "model-00002-of-00010.safetensors",
+    "language_model.model.layers.5.self_attn.q_proj.weight": "model-00002-of-00010.safetensors",
+    "language_model.model.layers.5.self_attn.v_proj.weight": "model-00002-of-00010.safetensors",
+    "language_model.model.layers.6.input_layernorm.weight": "model-00003-of-00010.safetensors",
+    "language_model.model.layers.6.mlp.down_proj.weight": "model-00003-of-00010.safetensors",
+    "language_model.model.layers.6.mlp.gate_proj.weight": "model-00002-of-00010.safetensors",
+    "language_model.model.layers.6.mlp.up_proj.weight": "model-00002-of-00010.safetensors",
+    "language_model.model.layers.6.post_attention_layernorm.weight": "model-00003-of-00010.safetensors",
+    "language_model.model.layers.6.self_attn.k_proj.weight": "model-00002-of-00010.safetensors",
+    "language_model.model.layers.6.self_attn.o_proj.weight": "model-00002-of-00010.safetensors",
+    "language_model.model.layers.6.self_attn.q_proj.weight": "model-00002-of-00010.safetensors",
+    "language_model.model.layers.6.self_attn.v_proj.weight": "model-00002-of-00010.safetensors",
+    "language_model.model.layers.7.input_layernorm.weight": "model-00003-of-00010.safetensors",
+    "language_model.model.layers.7.mlp.down_proj.weight": "model-00003-of-00010.safetensors",
+    "language_model.model.layers.7.mlp.gate_proj.weight": "model-00003-of-00010.safetensors",
+    "language_model.model.layers.7.mlp.up_proj.weight": "model-00003-of-00010.safetensors",
+    "language_model.model.layers.7.post_attention_layernorm.weight": "model-00003-of-00010.safetensors",
+    "language_model.model.layers.7.self_attn.k_proj.weight": "model-00003-of-00010.safetensors",
+    "language_model.model.layers.7.self_attn.o_proj.weight": "model-00003-of-00010.safetensors",
+    "language_model.model.layers.7.self_attn.q_proj.weight": "model-00003-of-00010.safetensors",
+    "language_model.model.layers.7.self_attn.v_proj.weight": "model-00003-of-00010.safetensors",
+    "language_model.model.layers.8.input_layernorm.weight": "model-00003-of-00010.safetensors",
+    "language_model.model.layers.8.mlp.down_proj.weight": "model-00003-of-00010.safetensors",
+    "language_model.model.layers.8.mlp.gate_proj.weight": "model-00003-of-00010.safetensors",
+    "language_model.model.layers.8.mlp.up_proj.weight": "model-00003-of-00010.safetensors",
+    "language_model.model.layers.8.post_attention_layernorm.weight": "model-00003-of-00010.safetensors",
+    "language_model.model.layers.8.self_attn.k_proj.weight": "model-00003-of-00010.safetensors",
+    "language_model.model.layers.8.self_attn.o_proj.weight": "model-00003-of-00010.safetensors",
+    "language_model.model.layers.8.self_attn.q_proj.weight": "model-00003-of-00010.safetensors",
+    "language_model.model.layers.8.self_attn.v_proj.weight": "model-00003-of-00010.safetensors",
+    "language_model.model.layers.9.input_layernorm.weight": "model-00003-of-00010.safetensors",
+    "language_model.model.layers.9.mlp.down_proj.weight": "model-00003-of-00010.safetensors",
+    "language_model.model.layers.9.mlp.gate_proj.weight": "model-00003-of-00010.safetensors",
+    "language_model.model.layers.9.mlp.up_proj.weight": "model-00003-of-00010.safetensors",
+    "language_model.model.layers.9.post_attention_layernorm.weight": "model-00003-of-00010.safetensors",
+    "language_model.model.layers.9.self_attn.k_proj.weight": "model-00003-of-00010.safetensors",
+    "language_model.model.layers.9.self_attn.o_proj.weight": "model-00003-of-00010.safetensors",
+    "language_model.model.layers.9.self_attn.q_proj.weight": "model-00003-of-00010.safetensors",
+    "language_model.model.layers.9.self_attn.v_proj.weight": "model-00003-of-00010.safetensors",
+    "language_model.model.norm.weight": "model-00010-of-00010.safetensors",
+    "multi_modal_projector.linear_1.weight": "model-00001-of-00010.safetensors",
+    "multi_modal_projector.linear_2.weight": "model-00001-of-00010.safetensors",
+    "multi_modal_projector.norm.weight": "model-00001-of-00010.safetensors",
+    "multi_modal_projector.patch_merger.merging_layer.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.ln_pre.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.patch_conv.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.0.attention.k_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.0.attention.o_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.0.attention.q_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.0.attention.v_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.0.attention_norm.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.0.feed_forward.down_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.0.feed_forward.gate_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.0.feed_forward.up_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.0.ffn_norm.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.1.attention.k_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.1.attention.o_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.1.attention.q_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.1.attention.v_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.1.attention_norm.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.1.feed_forward.down_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.1.feed_forward.gate_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.1.feed_forward.up_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.1.ffn_norm.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.10.attention.k_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.10.attention.o_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.10.attention.q_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.10.attention.v_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.10.attention_norm.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.10.feed_forward.down_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.10.feed_forward.gate_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.10.feed_forward.up_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.10.ffn_norm.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.11.attention.k_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.11.attention.o_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.11.attention.q_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.11.attention.v_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.11.attention_norm.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.11.feed_forward.down_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.11.feed_forward.gate_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.11.feed_forward.up_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.11.ffn_norm.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.12.attention.k_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.12.attention.o_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.12.attention.q_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.12.attention.v_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.12.attention_norm.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.12.feed_forward.down_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.12.feed_forward.gate_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.12.feed_forward.up_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.12.ffn_norm.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.13.attention.k_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.13.attention.o_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.13.attention.q_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.13.attention.v_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.13.attention_norm.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.13.feed_forward.down_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.13.feed_forward.gate_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.13.feed_forward.up_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.13.ffn_norm.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.14.attention.k_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.14.attention.o_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.14.attention.q_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.14.attention.v_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.14.attention_norm.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.14.feed_forward.down_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.14.feed_forward.gate_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.14.feed_forward.up_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.14.ffn_norm.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.15.attention.k_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.15.attention.o_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.15.attention.q_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.15.attention.v_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.15.attention_norm.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.15.feed_forward.down_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.15.feed_forward.gate_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.15.feed_forward.up_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.15.ffn_norm.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.16.attention.k_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.16.attention.o_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.16.attention.q_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.16.attention.v_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.16.attention_norm.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.16.feed_forward.down_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.16.feed_forward.gate_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.16.feed_forward.up_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.16.ffn_norm.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.17.attention.k_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.17.attention.o_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.17.attention.q_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.17.attention.v_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.17.attention_norm.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.17.feed_forward.down_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.17.feed_forward.gate_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.17.feed_forward.up_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.17.ffn_norm.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.18.attention.k_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.18.attention.o_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.18.attention.q_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.18.attention.v_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.18.attention_norm.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.18.feed_forward.down_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.18.feed_forward.gate_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.18.feed_forward.up_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.18.ffn_norm.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.19.attention.k_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.19.attention.o_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.19.attention.q_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.19.attention.v_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.19.attention_norm.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.19.feed_forward.down_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.19.feed_forward.gate_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.19.feed_forward.up_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.19.ffn_norm.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.2.attention.k_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.2.attention.o_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.2.attention.q_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.2.attention.v_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.2.attention_norm.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.2.feed_forward.down_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.2.feed_forward.gate_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.2.feed_forward.up_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.2.ffn_norm.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.20.attention.k_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.20.attention.o_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.20.attention.q_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.20.attention.v_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.20.attention_norm.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.20.feed_forward.down_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.20.feed_forward.gate_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.20.feed_forward.up_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.20.ffn_norm.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.21.attention.k_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.21.attention.o_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.21.attention.q_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.21.attention.v_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.21.attention_norm.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.21.feed_forward.down_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.21.feed_forward.gate_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.21.feed_forward.up_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.21.ffn_norm.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.22.attention.k_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.22.attention.o_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.22.attention.q_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.22.attention.v_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.22.attention_norm.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.22.feed_forward.down_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.22.feed_forward.gate_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.22.feed_forward.up_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.22.ffn_norm.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.23.attention.k_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.23.attention.o_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.23.attention.q_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.23.attention.v_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.23.attention_norm.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.23.feed_forward.down_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.23.feed_forward.gate_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.23.feed_forward.up_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.23.ffn_norm.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.3.attention.k_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.3.attention.o_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.3.attention.q_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.3.attention.v_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.3.attention_norm.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.3.feed_forward.down_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.3.feed_forward.gate_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.3.feed_forward.up_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.3.ffn_norm.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.4.attention.k_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.4.attention.o_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.4.attention.q_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.4.attention.v_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.4.attention_norm.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.4.feed_forward.down_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.4.feed_forward.gate_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.4.feed_forward.up_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.4.ffn_norm.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.5.attention.k_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.5.attention.o_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.5.attention.q_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.5.attention.v_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.5.attention_norm.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.5.feed_forward.down_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.5.feed_forward.gate_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.5.feed_forward.up_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.5.ffn_norm.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.6.attention.k_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.6.attention.o_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.6.attention.q_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.6.attention.v_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.6.attention_norm.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.6.feed_forward.down_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.6.feed_forward.gate_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.6.feed_forward.up_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.6.ffn_norm.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.7.attention.k_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.7.attention.o_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.7.attention.q_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.7.attention.v_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.7.attention_norm.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.7.feed_forward.down_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.7.feed_forward.gate_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.7.feed_forward.up_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.7.ffn_norm.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.8.attention.k_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.8.attention.o_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.8.attention.q_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.8.attention.v_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.8.attention_norm.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.8.feed_forward.down_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.8.feed_forward.gate_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.8.feed_forward.up_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.8.ffn_norm.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.9.attention.k_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.9.attention.o_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.9.attention.q_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.9.attention.v_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.9.attention_norm.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.9.feed_forward.down_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.9.feed_forward.gate_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.9.feed_forward.up_proj.weight": "model-00001-of-00010.safetensors",
+    "vision_tower.transformer.layers.9.ffn_norm.weight": "model-00001-of-00010.safetensors"
+  }
+}

tokenizer/chat_template.jinja

@@ -0,0 +1,51 @@
+{%- set today = strftime_now("%Y-%m-%d") %}
+{%- set default_system_message = "You are Mistral Small 3, a Large Language Model (LLM) created by Mistral AI, a French startup headquartered in Paris.\nYour knowledge base was last updated on 2023-10-01. The current date is " + today + ".\n\nWhen you're not sure about some information, you say that you don't have the information and don't make up anything.\nIf the user's question is not clear, ambiguous, or does not provide enough context for you to accurately answer the question, you do not try to answer it right away and you rather ask the user to clarify their request (e.g. \"What are some good restaurants around me?\" => \"Where are you?\" or \"When is the next flight to Tokyo\" => \"Where do you travel from?\")" %}
+
+{{- bos_token }}
+
+{%- if messages[0]['role'] == 'system' %}
+    {%- if messages[0]['content'] is string %}
+        {%- set system_message = messages[0]['content'] %}
+    {%- else %}
+        {%- set system_message = messages[0]['content'][0]['text'] %}
+    {%- endif %}
+    {%- set loop_messages = messages[1:] %}
+{%- else %}
+    {%- set system_message = default_system_message %}
+    {%- set loop_messages = messages %}
+{%- endif %}
+{{- '[SYSTEM_PROMPT]' + system_message + '[/SYSTEM_PROMPT]' }}
+
+{%- for message in loop_messages %}
+    {%- if message['role'] == 'user' %}
+        {%- if message['content'] is string %}
+            {{- '[INST]' + message['content'] + '[/INST]' }}
+        {%- else %}
+            {{- '[INST]' }}
+            {%- for block in message['content'] %}
+                {%- if block['type'] == 'text' %}
+                    {{- block['text'] }}
+                {%- elif block['type'] in ['image', 'image_url'] %}
+                    {{- '[IMG]' }}
+                {%- else %}
+                    {{- raise_exception('Only text and image blocks are supported in message content!') }}
+                {%- endif %}
+            {%- endfor %}
+            {{- '[/INST]' }}
+        {%- endif %}
+    {%- elif message['role'] == 'system' %}
+        {%- if message['content'] is string %}
+            {{- '[SYSTEM_PROMPT]' + message['content'] + '[/SYSTEM_PROMPT]' }}
+        {%- else %}
+            {{- '[SYSTEM_PROMPT]' + message['content'][0]['text'] + '[/SYSTEM_PROMPT]' }}
+        {%- endif %}
+    {%- elif message['role'] == 'assistant' %}
+        {%- if message['content'] is string %}
+            {{- message['content'] + eos_token }}
+        {%- else %}
+            {{- message['content'][0]['text'] + eos_token }}
+        {%- endif %}
+    {%- else %}
+        {{- raise_exception('Only user, system and assistant roles are supported!') }}
+    {%- endif %}
+{%- endfor %}

tokenizer/preprocessor_config.json

@@ -0,0 +1,34 @@
+{
+  "crop_size": null,
+  "data_format": "channels_first",
+  "default_to_square": true,
+  "device": null,
+  "disable_grouping": null,
+  "do_center_crop": null,
+  "do_convert_rgb": true,
+  "do_normalize": true,
+  "do_pad": null,
+  "do_rescale": true,
+  "do_resize": true,
+  "image_mean": [
+    0.48145466,
+    0.4578275,
+    0.40821073
+  ],
+  "image_processor_type": "PixtralImageProcessorFast",
+  "image_std": [
+    0.26862954,
+    0.26130258,
+    0.27577711
+  ],
+  "input_data_format": null,
+  "pad_size": null,
+  "patch_size": 14,
+  "processor_class": "PixtralProcessor",
+  "resample": 3,
+  "rescale_factor": 0.00392156862745098,
+  "return_tensors": null,
+  "size": {
+    "longest_edge": 1540
+  }
+}

tokenizer/processor_config.json

@@ -0,0 +1,8 @@
+{
+  "image_break_token": "[IMG_BREAK]",
+  "image_end_token": "[IMG_END]",
+  "image_token": "[IMG]",
+  "patch_size": 14,
+  "processor_class": "PixtralProcessor",
+  "spatial_merge_size": 2
+}

tokenizer/special_tokens_map.json

@@ -0,0 +1,1032 @@
+{
+  "additional_special_tokens": [
+    "<unk>",
+    "<s>",
+    "</s>",
+    "[INST]",
+    "[/INST]",
+    "[AVAILABLE_TOOLS]",
+    "[/AVAILABLE_TOOLS]",
+    "[TOOL_RESULTS]",
+    "[/TOOL_RESULTS]",
+    "[TOOL_CALLS]",
+    "[IMG]",
+    "<pad>",
+    "[IMG_BREAK]",
+    "[IMG_END]",
+    "[PREFIX]",
+    "[MIDDLE]",
+    "[SUFFIX]",
+    "[SYSTEM_PROMPT]",
+    "[/SYSTEM_PROMPT]",
+    "[TOOL_CONTENT]",
+    "<SPECIAL_20>",
+    "<SPECIAL_21>",
+    "<SPECIAL_22>",
+    "<SPECIAL_23>",
+    "<SPECIAL_24>",
+    "<SPECIAL_25>",
+    "<SPECIAL_26>",
+    "<SPECIAL_27>",
+    "<SPECIAL_28>",
+    "<SPECIAL_29>",
+    "<SPECIAL_30>",
+    "<SPECIAL_31>",
+    "<SPECIAL_32>",
+    "<SPECIAL_33>",
+    "<SPECIAL_34>",
+    "<SPECIAL_35>",
+    "<SPECIAL_36>",
+    "<SPECIAL_37>",
+    "<SPECIAL_38>",
+    "<SPECIAL_39>",
+    "<SPECIAL_40>",
+    "<SPECIAL_41>",
+    "<SPECIAL_42>",
+    "<SPECIAL_43>",
+    "<SPECIAL_44>",
+    "<SPECIAL_45>",
+    "<SPECIAL_46>",
+    "<SPECIAL_47>",
+    "<SPECIAL_48>",
+    "<SPECIAL_49>",
+    "<SPECIAL_50>",
+    "<SPECIAL_51>",
+    "<SPECIAL_52>",
+    "<SPECIAL_53>",
+    "<SPECIAL_54>",
+    "<SPECIAL_55>",
+    "<SPECIAL_56>",
+    "<SPECIAL_57>",
+    "<SPECIAL_58>",
+    "<SPECIAL_59>",
+    "<SPECIAL_60>",
+    "<SPECIAL_61>",
+    "<SPECIAL_62>",
+    "<SPECIAL_63>",
+    "<SPECIAL_64>",
+    "<SPECIAL_65>",
+    "<SPECIAL_66>",
+    "<SPECIAL_67>",
+    "<SPECIAL_68>",
+    "<SPECIAL_69>",
+    "<SPECIAL_70>",
+    "<SPECIAL_71>",
+    "<SPECIAL_72>",
+    "<SPECIAL_73>",
+    "<SPECIAL_74>",
+    "<SPECIAL_75>",
+    "<SPECIAL_76>",
+    "<SPECIAL_77>",
+    "<SPECIAL_78>",
+    "<SPECIAL_79>",
+    "<SPECIAL_80>",
+    "<SPECIAL_81>",
+    "<SPECIAL_82>",
+    "<SPECIAL_83>",
+    "<SPECIAL_84>",
+    "<SPECIAL_85>",
+    "<SPECIAL_86>",
+    "<SPECIAL_87>",
+    "<SPECIAL_88>",
+    "<SPECIAL_89>",
+    "<SPECIAL_90>",
+    "<SPECIAL_91>",
+    "<SPECIAL_92>",
+    "<SPECIAL_93>",
+    "<SPECIAL_94>",
+    "<SPECIAL_95>",
+    "<SPECIAL_96>",
+    "<SPECIAL_97>",
+    "<SPECIAL_98>",
+    "<SPECIAL_99>",
+    "<SPECIAL_100>",
+    "<SPECIAL_101>",
+    "<SPECIAL_102>",
+    "<SPECIAL_103>",
+    "<SPECIAL_104>",
+    "<SPECIAL_105>",
+    "<SPECIAL_106>",
+    "<SPECIAL_107>",
+    "<SPECIAL_108>",
+    "<SPECIAL_109>",
+    "<SPECIAL_110>",
+    "<SPECIAL_111>",
+    "<SPECIAL_112>",
+    "<SPECIAL_113>",
+    "<SPECIAL_114>",
+    "<SPECIAL_115>",
+    "<SPECIAL_116>",
+    "<SPECIAL_117>",
+    "<SPECIAL_118>",
+    "<SPECIAL_119>",
+    "<SPECIAL_120>",
+    "<SPECIAL_121>",
+    "<SPECIAL_122>",
+    "<SPECIAL_123>",
+    "<SPECIAL_124>",
+    "<SPECIAL_125>",
+    "<SPECIAL_126>",
+    "<SPECIAL_127>",
+    "<SPECIAL_128>",
+    "<SPECIAL_129>",
+    "<SPECIAL_130>",
+    "<SPECIAL_131>",
+    "<SPECIAL_132>",
+    "<SPECIAL_133>",
+    "<SPECIAL_134>",
+    "<SPECIAL_135>",
+    "<SPECIAL_136>",
+    "<SPECIAL_137>",
+    "<SPECIAL_138>",
+    "<SPECIAL_139>",
+    "<SPECIAL_140>",
+    "<SPECIAL_141>",
+    "<SPECIAL_142>",
+    "<SPECIAL_143>",
+    "<SPECIAL_144>",
+    "<SPECIAL_145>",
+    "<SPECIAL_146>",
+    "<SPECIAL_147>",
+    "<SPECIAL_148>",
+    "<SPECIAL_149>",
+    "<SPECIAL_150>",
+    "<SPECIAL_151>",
+    "<SPECIAL_152>",
+    "<SPECIAL_153>",
+    "<SPECIAL_154>",
+    "<SPECIAL_155>",
+    "<SPECIAL_156>",
+    "<SPECIAL_157>",
+    "<SPECIAL_158>",
+    "<SPECIAL_159>",
+    "<SPECIAL_160>",
+    "<SPECIAL_161>",
+    "<SPECIAL_162>",
+    "<SPECIAL_163>",
+    "<SPECIAL_164>",
+    "<SPECIAL_165>",
+    "<SPECIAL_166>",
+    "<SPECIAL_167>",
+    "<SPECIAL_168>",
+    "<SPECIAL_169>",
+    "<SPECIAL_170>",
+    "<SPECIAL_171>",
+    "<SPECIAL_172>",
+    "<SPECIAL_173>",
+    "<SPECIAL_174>",
+    "<SPECIAL_175>",
+    "<SPECIAL_176>",
+    "<SPECIAL_177>",
+    "<SPECIAL_178>",
+    "<SPECIAL_179>",
+    "<SPECIAL_180>",
+    "<SPECIAL_181>",
+    "<SPECIAL_182>",
+    "<SPECIAL_183>",
+    "<SPECIAL_184>",
+    "<SPECIAL_185>",
+    "<SPECIAL_186>",
+    "<SPECIAL_187>",
+    "<SPECIAL_188>",
+    "<SPECIAL_189>",
+    "<SPECIAL_190>",
+    "<SPECIAL_191>",
+    "<SPECIAL_192>",
+    "<SPECIAL_193>",
+    "<SPECIAL_194>",
+    "<SPECIAL_195>",
+    "<SPECIAL_196>",
+    "<SPECIAL_197>",
+    "<SPECIAL_198>",
+    "<SPECIAL_199>",
+    "<SPECIAL_200>",
+    "<SPECIAL_201>",
+    "<SPECIAL_202>",
+    "<SPECIAL_203>",
+    "<SPECIAL_204>",
+    "<SPECIAL_205>",
+    "<SPECIAL_206>",
+    "<SPECIAL_207>",
+    "<SPECIAL_208>",
+    "<SPECIAL_209>",
+    "<SPECIAL_210>",
+    "<SPECIAL_211>",
+    "<SPECIAL_212>",
+    "<SPECIAL_213>",
+    "<SPECIAL_214>",
+    "<SPECIAL_215>",
+    "<SPECIAL_216>",
+    "<SPECIAL_217>",
+    "<SPECIAL_218>",
+    "<SPECIAL_219>",
+    "<SPECIAL_220>",
+    "<SPECIAL_221>",
+    "<SPECIAL_222>",
+    "<SPECIAL_223>",
+    "<SPECIAL_224>",
+    "<SPECIAL_225>",
+    "<SPECIAL_226>",
+    "<SPECIAL_227>",
+    "<SPECIAL_228>",
+    "<SPECIAL_229>",
+    "<SPECIAL_230>",
+    "<SPECIAL_231>",
+    "<SPECIAL_232>",
+    "<SPECIAL_233>",
+    "<SPECIAL_234>",
+    "<SPECIAL_235>",
+    "<SPECIAL_236>",
+    "<SPECIAL_237>",
+    "<SPECIAL_238>",
+    "<SPECIAL_239>",
+    "<SPECIAL_240>",
+    "<SPECIAL_241>",
+    "<SPECIAL_242>",
+    "<SPECIAL_243>",
+    "<SPECIAL_244>",
+    "<SPECIAL_245>",
+    "<SPECIAL_246>",
+    "<SPECIAL_247>",
+    "<SPECIAL_248>",
+    "<SPECIAL_249>",
+    "<SPECIAL_250>",
+    "<SPECIAL_251>",
+    "<SPECIAL_252>",
+    "<SPECIAL_253>",
+    "<SPECIAL_254>",
+    "<SPECIAL_255>",
+    "<SPECIAL_256>",
+    "<SPECIAL_257>",
+    "<SPECIAL_258>",
+    "<SPECIAL_259>",
+    "<SPECIAL_260>",
+    "<SPECIAL_261>",
+    "<SPECIAL_262>",
+    "<SPECIAL_263>",
+    "<SPECIAL_264>",
+    "<SPECIAL_265>",
+    "<SPECIAL_266>",
+    "<SPECIAL_267>",
+    "<SPECIAL_268>",
+    "<SPECIAL_269>",
+    "<SPECIAL_270>",
+    "<SPECIAL_271>",
+    "<SPECIAL_272>",
+    "<SPECIAL_273>",
+    "<SPECIAL_274>",
+    "<SPECIAL_275>",
+    "<SPECIAL_276>",
+    "<SPECIAL_277>",
+    "<SPECIAL_278>",
+    "<SPECIAL_279>",
+    "<SPECIAL_280>",
+    "<SPECIAL_281>",
+    "<SPECIAL_282>",
+    "<SPECIAL_283>",
+    "<SPECIAL_284>",
+    "<SPECIAL_285>",
+    "<SPECIAL_286>",
+    "<SPECIAL_287>",
+    "<SPECIAL_288>",
+    "<SPECIAL_289>",
+    "<SPECIAL_290>",
+    "<SPECIAL_291>",
+    "<SPECIAL_292>",
+    "<SPECIAL_293>",
+    "<SPECIAL_294>",
+    "<SPECIAL_295>",
+    "<SPECIAL_296>",
+    "<SPECIAL_297>",
+    "<SPECIAL_298>",
+    "<SPECIAL_299>",
+    "<SPECIAL_300>",
+    "<SPECIAL_301>",
+    "<SPECIAL_302>",
+    "<SPECIAL_303>",
+    "<SPECIAL_304>",
+    "<SPECIAL_305>",
+    "<SPECIAL_306>",
+    "<SPECIAL_307>",
+    "<SPECIAL_308>",
+    "<SPECIAL_309>",
+    "<SPECIAL_310>",
+    "<SPECIAL_311>",
+    "<SPECIAL_312>",
+    "<SPECIAL_313>",
+    "<SPECIAL_314>",
+    "<SPECIAL_315>",
+    "<SPECIAL_316>",
+    "<SPECIAL_317>",
+    "<SPECIAL_318>",
+    "<SPECIAL_319>",
+    "<SPECIAL_320>",
+    "<SPECIAL_321>",
+    "<SPECIAL_322>",
+    "<SPECIAL_323>",
+    "<SPECIAL_324>",
+    "<SPECIAL_325>",
+    "<SPECIAL_326>",
+    "<SPECIAL_327>",
+    "<SPECIAL_328>",
+    "<SPECIAL_329>",
+    "<SPECIAL_330>",
+    "<SPECIAL_331>",
+    "<SPECIAL_332>",
+    "<SPECIAL_333>",
+    "<SPECIAL_334>",
+    "<SPECIAL_335>",
+    "<SPECIAL_336>",
+    "<SPECIAL_337>",
+    "<SPECIAL_338>",
+    "<SPECIAL_339>",
+    "<SPECIAL_340>",
+    "<SPECIAL_341>",
+    "<SPECIAL_342>",
+    "<SPECIAL_343>",
+    "<SPECIAL_344>",
+    "<SPECIAL_345>",
+    "<SPECIAL_346>",
+    "<SPECIAL_347>",
+    "<SPECIAL_348>",
+    "<SPECIAL_349>",
+    "<SPECIAL_350>",
+    "<SPECIAL_351>",
+    "<SPECIAL_352>",
+    "<SPECIAL_353>",
+    "<SPECIAL_354>",
+    "<SPECIAL_355>",
+    "<SPECIAL_356>",
+    "<SPECIAL_357>",
+    "<SPECIAL_358>",
+    "<SPECIAL_359>",
+    "<SPECIAL_360>",
+    "<SPECIAL_361>",
+    "<SPECIAL_362>",
+    "<SPECIAL_363>",
+    "<SPECIAL_364>",
+    "<SPECIAL_365>",
+    "<SPECIAL_366>",
+    "<SPECIAL_367>",
+    "<SPECIAL_368>",
+    "<SPECIAL_369>",
+    "<SPECIAL_370>",
+    "<SPECIAL_371>",
+    "<SPECIAL_372>",
+    "<SPECIAL_373>",
+    "<SPECIAL_374>",
+    "<SPECIAL_375>",
+    "<SPECIAL_376>",
+    "<SPECIAL_377>",
+    "<SPECIAL_378>",
+    "<SPECIAL_379>",
+    "<SPECIAL_380>",
+    "<SPECIAL_381>",
+    "<SPECIAL_382>",
+    "<SPECIAL_383>",
+    "<SPECIAL_384>",
+    "<SPECIAL_385>",
+    "<SPECIAL_386>",
+    "<SPECIAL_387>",
+    "<SPECIAL_388>",
+    "<SPECIAL_389>",
+    "<SPECIAL_390>",
+    "<SPECIAL_391>",
+    "<SPECIAL_392>",
+    "<SPECIAL_393>",
+    "<SPECIAL_394>",
+    "<SPECIAL_395>",
+    "<SPECIAL_396>",
+    "<SPECIAL_397>",
+    "<SPECIAL_398>",
+    "<SPECIAL_399>",
+    "<SPECIAL_400>",
+    "<SPECIAL_401>",
+    "<SPECIAL_402>",
+    "<SPECIAL_403>",
+    "<SPECIAL_404>",
+    "<SPECIAL_405>",
+    "<SPECIAL_406>",
+    "<SPECIAL_407>",
+    "<SPECIAL_408>",
+    "<SPECIAL_409>",
+    "<SPECIAL_410>",
+    "<SPECIAL_411>",
+    "<SPECIAL_412>",
+    "<SPECIAL_413>",
+    "<SPECIAL_414>",
+    "<SPECIAL_415>",
+    "<SPECIAL_416>",
+    "<SPECIAL_417>",
+    "<SPECIAL_418>",
+    "<SPECIAL_419>",
+    "<SPECIAL_420>",
+    "<SPECIAL_421>",
+    "<SPECIAL_422>",
+    "<SPECIAL_423>",
+    "<SPECIAL_424>",
+    "<SPECIAL_425>",
+    "<SPECIAL_426>",
+    "<SPECIAL_427>",
+    "<SPECIAL_428>",
+    "<SPECIAL_429>",
+    "<SPECIAL_430>",
+    "<SPECIAL_431>",
+    "<SPECIAL_432>",
+    "<SPECIAL_433>",
+    "<SPECIAL_434>",
+    "<SPECIAL_435>",
+    "<SPECIAL_436>",
+    "<SPECIAL_437>",
+    "<SPECIAL_438>",
+    "<SPECIAL_439>",
+    "<SPECIAL_440>",
+    "<SPECIAL_441>",
+    "<SPECIAL_442>",
+    "<SPECIAL_443>",
+    "<SPECIAL_444>",
+    "<SPECIAL_445>",
+    "<SPECIAL_446>",
+    "<SPECIAL_447>",
+    "<SPECIAL_448>",
+    "<SPECIAL_449>",
+    "<SPECIAL_450>",
+    "<SPECIAL_451>",
+    "<SPECIAL_452>",
+    "<SPECIAL_453>",
+    "<SPECIAL_454>",
+    "<SPECIAL_455>",
+    "<SPECIAL_456>",
+    "<SPECIAL_457>",
+    "<SPECIAL_458>",
+    "<SPECIAL_459>",
+    "<SPECIAL_460>",
+    "<SPECIAL_461>",
+    "<SPECIAL_462>",
+    "<SPECIAL_463>",
+    "<SPECIAL_464>",
+    "<SPECIAL_465>",
+    "<SPECIAL_466>",
+    "<SPECIAL_467>",
+    "<SPECIAL_468>",
+    "<SPECIAL_469>",
+    "<SPECIAL_470>",
+    "<SPECIAL_471>",
+    "<SPECIAL_472>",
+    "<SPECIAL_473>",
+    "<SPECIAL_474>",
+    "<SPECIAL_475>",
+    "<SPECIAL_476>",
+    "<SPECIAL_477>",
+    "<SPECIAL_478>",
+    "<SPECIAL_479>",
+    "<SPECIAL_480>",
+    "<SPECIAL_481>",
+    "<SPECIAL_482>",
+    "<SPECIAL_483>",
+    "<SPECIAL_484>",
+    "<SPECIAL_485>",
+    "<SPECIAL_486>",
+    "<SPECIAL_487>",
+    "<SPECIAL_488>",
+    "<SPECIAL_489>",
+    "<SPECIAL_490>",
+    "<SPECIAL_491>",
+    "<SPECIAL_492>",
+    "<SPECIAL_493>",
+    "<SPECIAL_494>",
+    "<SPECIAL_495>",
+    "<SPECIAL_496>",
+    "<SPECIAL_497>",
+    "<SPECIAL_498>",
+    "<SPECIAL_499>",
+    "<SPECIAL_500>",
+    "<SPECIAL_501>",
+    "<SPECIAL_502>",
+    "<SPECIAL_503>",
+    "<SPECIAL_504>",
+    "<SPECIAL_505>",
+    "<SPECIAL_506>",
+    "<SPECIAL_507>",
+    "<SPECIAL_508>",
+    "<SPECIAL_509>",
+    "<SPECIAL_510>",
+    "<SPECIAL_511>",
+    "<SPECIAL_512>",
+    "<SPECIAL_513>",
+    "<SPECIAL_514>",
+    "<SPECIAL_515>",
+    "<SPECIAL_516>",
+    "<SPECIAL_517>",
+    "<SPECIAL_518>",
+    "<SPECIAL_519>",
+    "<SPECIAL_520>",
+    "<SPECIAL_521>",
+    "<SPECIAL_522>",
+    "<SPECIAL_523>",
+    "<SPECIAL_524>",
+    "<SPECIAL_525>",
+    "<SPECIAL_526>",
+    "<SPECIAL_527>",
+    "<SPECIAL_528>",
+    "<SPECIAL_529>",
+    "<SPECIAL_530>",
+    "<SPECIAL_531>",
+    "<SPECIAL_532>",
+    "<SPECIAL_533>",
+    "<SPECIAL_534>",
+    "<SPECIAL_535>",
+    "<SPECIAL_536>",
+    "<SPECIAL_537>",
+    "<SPECIAL_538>",
+    "<SPECIAL_539>",
+    "<SPECIAL_540>",
+    "<SPECIAL_541>",
+    "<SPECIAL_542>",
+    "<SPECIAL_543>",
+    "<SPECIAL_544>",
+    "<SPECIAL_545>",
+    "<SPECIAL_546>",
+    "<SPECIAL_547>",
+    "<SPECIAL_548>",
+    "<SPECIAL_549>",
+    "<SPECIAL_550>",
+    "<SPECIAL_551>",
+    "<SPECIAL_552>",
+    "<SPECIAL_553>",
+    "<SPECIAL_554>",
+    "<SPECIAL_555>",
+    "<SPECIAL_556>",
+    "<SPECIAL_557>",
+    "<SPECIAL_558>",
+    "<SPECIAL_559>",
+    "<SPECIAL_560>",
+    "<SPECIAL_561>",
+    "<SPECIAL_562>",
+    "<SPECIAL_563>",
+    "<SPECIAL_564>",
+    "<SPECIAL_565>",
+    "<SPECIAL_566>",
+    "<SPECIAL_567>",
+    "<SPECIAL_568>",
+    "<SPECIAL_569>",
+    "<SPECIAL_570>",
+    "<SPECIAL_571>",
+    "<SPECIAL_572>",
+    "<SPECIAL_573>",
+    "<SPECIAL_574>",
+    "<SPECIAL_575>",
+    "<SPECIAL_576>",
+    "<SPECIAL_577>",
+    "<SPECIAL_578>",
+    "<SPECIAL_579>",
+    "<SPECIAL_580>",
+    "<SPECIAL_581>",
+    "<SPECIAL_582>",
+    "<SPECIAL_583>",
+    "<SPECIAL_584>",
+    "<SPECIAL_585>",
+    "<SPECIAL_586>",
+    "<SPECIAL_587>",
+    "<SPECIAL_588>",
+    "<SPECIAL_589>",
+    "<SPECIAL_590>",
+    "<SPECIAL_591>",
+    "<SPECIAL_592>",
+    "<SPECIAL_593>",
+    "<SPECIAL_594>",
+    "<SPECIAL_595>",
+    "<SPECIAL_596>",
+    "<SPECIAL_597>",
+    "<SPECIAL_598>",
+    "<SPECIAL_599>",
+    "<SPECIAL_600>",
+    "<SPECIAL_601>",
+    "<SPECIAL_602>",
+    "<SPECIAL_603>",
+    "<SPECIAL_604>",
+    "<SPECIAL_605>",
+    "<SPECIAL_606>",
+    "<SPECIAL_607>",
+    "<SPECIAL_608>",
+    "<SPECIAL_609>",
+    "<SPECIAL_610>",
+    "<SPECIAL_611>",
+    "<SPECIAL_612>",
+    "<SPECIAL_613>",
+    "<SPECIAL_614>",
+    "<SPECIAL_615>",
+    "<SPECIAL_616>",
+    "<SPECIAL_617>",
+    "<SPECIAL_618>",
+    "<SPECIAL_619>",
+    "<SPECIAL_620>",
+    "<SPECIAL_621>",
+    "<SPECIAL_622>",
+    "<SPECIAL_623>",
+    "<SPECIAL_624>",
+    "<SPECIAL_625>",
+    "<SPECIAL_626>",
+    "<SPECIAL_627>",
+    "<SPECIAL_628>",
+    "<SPECIAL_629>",
+    "<SPECIAL_630>",
+    "<SPECIAL_631>",
+    "<SPECIAL_632>",
+    "<SPECIAL_633>",
+    "<SPECIAL_634>",
+    "<SPECIAL_635>",
+    "<SPECIAL_636>",
+    "<SPECIAL_637>",
+    "<SPECIAL_638>",
+    "<SPECIAL_639>",
+    "<SPECIAL_640>",
+    "<SPECIAL_641>",
+    "<SPECIAL_642>",
+    "<SPECIAL_643>",
+    "<SPECIAL_644>",
+    "<SPECIAL_645>",
+    "<SPECIAL_646>",
+    "<SPECIAL_647>",
+    "<SPECIAL_648>",
+    "<SPECIAL_649>",
+    "<SPECIAL_650>",
+    "<SPECIAL_651>",
+    "<SPECIAL_652>",
+    "<SPECIAL_653>",
+    "<SPECIAL_654>",
+    "<SPECIAL_655>",
+    "<SPECIAL_656>",
+    "<SPECIAL_657>",
+    "<SPECIAL_658>",
+    "<SPECIAL_659>",
+    "<SPECIAL_660>",
+    "<SPECIAL_661>",
+    "<SPECIAL_662>",
+    "<SPECIAL_663>",
+    "<SPECIAL_664>",
+    "<SPECIAL_665>",
+    "<SPECIAL_666>",
+    "<SPECIAL_667>",
+    "<SPECIAL_668>",
+    "<SPECIAL_669>",
+    "<SPECIAL_670>",
+    "<SPECIAL_671>",
+    "<SPECIAL_672>",
+    "<SPECIAL_673>",
+    "<SPECIAL_674>",
+    "<SPECIAL_675>",
+    "<SPECIAL_676>",
+    "<SPECIAL_677>",
+    "<SPECIAL_678>",
+    "<SPECIAL_679>",
+    "<SPECIAL_680>",
+    "<SPECIAL_681>",
+    "<SPECIAL_682>",
+    "<SPECIAL_683>",
+    "<SPECIAL_684>",
+    "<SPECIAL_685>",
+    "<SPECIAL_686>",
+    "<SPECIAL_687>",
+    "<SPECIAL_688>",
+    "<SPECIAL_689>",
+    "<SPECIAL_690>",
+    "<SPECIAL_691>",
+    "<SPECIAL_692>",
+    "<SPECIAL_693>",
+    "<SPECIAL_694>",
+    "<SPECIAL_695>",
+    "<SPECIAL_696>",
+    "<SPECIAL_697>",
+    "<SPECIAL_698>",
+    "<SPECIAL_699>",
+    "<SPECIAL_700>",
+    "<SPECIAL_701>",
+    "<SPECIAL_702>",
+    "<SPECIAL_703>",
+    "<SPECIAL_704>",
+    "<SPECIAL_705>",
+    "<SPECIAL_706>",
+    "<SPECIAL_707>",
+    "<SPECIAL_708>",
+    "<SPECIAL_709>",
+    "<SPECIAL_710>",
+    "<SPECIAL_711>",
+    "<SPECIAL_712>",
+    "<SPECIAL_713>",
+    "<SPECIAL_714>",
+    "<SPECIAL_715>",
+    "<SPECIAL_716>",
+    "<SPECIAL_717>",
+    "<SPECIAL_718>",
+    "<SPECIAL_719>",
+    "<SPECIAL_720>",
+    "<SPECIAL_721>",
+    "<SPECIAL_722>",
+    "<SPECIAL_723>",
+    "<SPECIAL_724>",
+    "<SPECIAL_725>",
+    "<SPECIAL_726>",
+    "<SPECIAL_727>",
+    "<SPECIAL_728>",
+    "<SPECIAL_729>",
+    "<SPECIAL_730>",
+    "<SPECIAL_731>",
+    "<SPECIAL_732>",
+    "<SPECIAL_733>",
+    "<SPECIAL_734>",
+    "<SPECIAL_735>",
+    "<SPECIAL_736>",
+    "<SPECIAL_737>",
+    "<SPECIAL_738>",
+    "<SPECIAL_739>",
+    "<SPECIAL_740>",
+    "<SPECIAL_741>",
+    "<SPECIAL_742>",
+    "<SPECIAL_743>",
+    "<SPECIAL_744>",
+    "<SPECIAL_745>",
+    "<SPECIAL_746>",
+    "<SPECIAL_747>",
+    "<SPECIAL_748>",
+    "<SPECIAL_749>",
+    "<SPECIAL_750>",
+    "<SPECIAL_751>",
+    "<SPECIAL_752>",
+    "<SPECIAL_753>",
+    "<SPECIAL_754>",
+    "<SPECIAL_755>",
+    "<SPECIAL_756>",
+    "<SPECIAL_757>",
+    "<SPECIAL_758>",
+    "<SPECIAL_759>",
+    "<SPECIAL_760>",
+    "<SPECIAL_761>",
+    "<SPECIAL_762>",
+    "<SPECIAL_763>",
+    "<SPECIAL_764>",
+    "<SPECIAL_765>",
+    "<SPECIAL_766>",
+    "<SPECIAL_767>",
+    "<SPECIAL_768>",
+    "<SPECIAL_769>",
+    "<SPECIAL_770>",
+    "<SPECIAL_771>",
+    "<SPECIAL_772>",
+    "<SPECIAL_773>",
+    "<SPECIAL_774>",
+    "<SPECIAL_775>",
+    "<SPECIAL_776>",
+    "<SPECIAL_777>",
+    "<SPECIAL_778>",
+    "<SPECIAL_779>",
+    "<SPECIAL_780>",
+    "<SPECIAL_781>",
+    "<SPECIAL_782>",
+    "<SPECIAL_783>",
+    "<SPECIAL_784>",
+    "<SPECIAL_785>",
+    "<SPECIAL_786>",
+    "<SPECIAL_787>",
+    "<SPECIAL_788>",
+    "<SPECIAL_789>",
+    "<SPECIAL_790>",
+    "<SPECIAL_791>",
+    "<SPECIAL_792>",
+    "<SPECIAL_793>",
+    "<SPECIAL_794>",
+    "<SPECIAL_795>",
+    "<SPECIAL_796>",
+    "<SPECIAL_797>",
+    "<SPECIAL_798>",
+    "<SPECIAL_799>",
+    "<SPECIAL_800>",
+    "<SPECIAL_801>",
+    "<SPECIAL_802>",
+    "<SPECIAL_803>",
+    "<SPECIAL_804>",
+    "<SPECIAL_805>",
+    "<SPECIAL_806>",
+    "<SPECIAL_807>",
+    "<SPECIAL_808>",
+    "<SPECIAL_809>",
+    "<SPECIAL_810>",
+    "<SPECIAL_811>",
+    "<SPECIAL_812>",
+    "<SPECIAL_813>",
+    "<SPECIAL_814>",
+    "<SPECIAL_815>",
+    "<SPECIAL_816>",
+    "<SPECIAL_817>",
+    "<SPECIAL_818>",
+    "<SPECIAL_819>",
+    "<SPECIAL_820>",
+    "<SPECIAL_821>",
+    "<SPECIAL_822>",
+    "<SPECIAL_823>",
+    "<SPECIAL_824>",
+    "<SPECIAL_825>",
+    "<SPECIAL_826>",
+    "<SPECIAL_827>",
+    "<SPECIAL_828>",
+    "<SPECIAL_829>",
+    "<SPECIAL_830>",
+    "<SPECIAL_831>",
+    "<SPECIAL_832>",
+    "<SPECIAL_833>",
+    "<SPECIAL_834>",
+    "<SPECIAL_835>",
+    "<SPECIAL_836>",
+    "<SPECIAL_837>",
+    "<SPECIAL_838>",
+    "<SPECIAL_839>",
+    "<SPECIAL_840>",
+    "<SPECIAL_841>",
+    "<SPECIAL_842>",
+    "<SPECIAL_843>",
+    "<SPECIAL_844>",
+    "<SPECIAL_845>",
+    "<SPECIAL_846>",
+    "<SPECIAL_847>",
+    "<SPECIAL_848>",
+    "<SPECIAL_849>",
+    "<SPECIAL_850>",
+    "<SPECIAL_851>",
+    "<SPECIAL_852>",
+    "<SPECIAL_853>",
+    "<SPECIAL_854>",
+    "<SPECIAL_855>",
+    "<SPECIAL_856>",
+    "<SPECIAL_857>",
+    "<SPECIAL_858>",
+    "<SPECIAL_859>",
+    "<SPECIAL_860>",
+    "<SPECIAL_861>",
+    "<SPECIAL_862>",
+    "<SPECIAL_863>",
+    "<SPECIAL_864>",
+    "<SPECIAL_865>",
+    "<SPECIAL_866>",
+    "<SPECIAL_867>",
+    "<SPECIAL_868>",
+    "<SPECIAL_869>",
+    "<SPECIAL_870>",
+    "<SPECIAL_871>",
+    "<SPECIAL_872>",
+    "<SPECIAL_873>",
+    "<SPECIAL_874>",
+    "<SPECIAL_875>",
+    "<SPECIAL_876>",
+    "<SPECIAL_877>",
+    "<SPECIAL_878>",
+    "<SPECIAL_879>",
+    "<SPECIAL_880>",
+    "<SPECIAL_881>",
+    "<SPECIAL_882>",
+    "<SPECIAL_883>",
+    "<SPECIAL_884>",
+    "<SPECIAL_885>",
+    "<SPECIAL_886>",
+    "<SPECIAL_887>",
+    "<SPECIAL_888>",
+    "<SPECIAL_889>",
+    "<SPECIAL_890>",
+    "<SPECIAL_891>",
+    "<SPECIAL_892>",
+    "<SPECIAL_893>",
+    "<SPECIAL_894>",
+    "<SPECIAL_895>",
+    "<SPECIAL_896>",
+    "<SPECIAL_897>",
+    "<SPECIAL_898>",
+    "<SPECIAL_899>",
+    "<SPECIAL_900>",
+    "<SPECIAL_901>",
+    "<SPECIAL_902>",
+    "<SPECIAL_903>",
+    "<SPECIAL_904>",
+    "<SPECIAL_905>",
+    "<SPECIAL_906>",
+    "<SPECIAL_907>",
+    "<SPECIAL_908>",
+    "<SPECIAL_909>",
+    "<SPECIAL_910>",
+    "<SPECIAL_911>",
+    "<SPECIAL_912>",
+    "<SPECIAL_913>",
+    "<SPECIAL_914>",
+    "<SPECIAL_915>",
+    "<SPECIAL_916>",
+    "<SPECIAL_917>",
+    "<SPECIAL_918>",
+    "<SPECIAL_919>",
+    "<SPECIAL_920>",
+    "<SPECIAL_921>",
+    "<SPECIAL_922>",
+    "<SPECIAL_923>",
+    "<SPECIAL_924>",
+    "<SPECIAL_925>",
+    "<SPECIAL_926>",
+    "<SPECIAL_927>",
+    "<SPECIAL_928>",
+    "<SPECIAL_929>",
+    "<SPECIAL_930>",
+    "<SPECIAL_931>",
+    "<SPECIAL_932>",
+    "<SPECIAL_933>",
+    "<SPECIAL_934>",
+    "<SPECIAL_935>",
+    "<SPECIAL_936>",
+    "<SPECIAL_937>",
+    "<SPECIAL_938>",
+    "<SPECIAL_939>",
+    "<SPECIAL_940>",
+    "<SPECIAL_941>",
+    "<SPECIAL_942>",
+    "<SPECIAL_943>",
+    "<SPECIAL_944>",
+    "<SPECIAL_945>",
+    "<SPECIAL_946>",
+    "<SPECIAL_947>",
+    "<SPECIAL_948>",
+    "<SPECIAL_949>",
+    "<SPECIAL_950>",
+    "<SPECIAL_951>",
+    "<SPECIAL_952>",
+    "<SPECIAL_953>",
+    "<SPECIAL_954>",
+    "<SPECIAL_955>",
+    "<SPECIAL_956>",
+    "<SPECIAL_957>",
+    "<SPECIAL_958>",
+    "<SPECIAL_959>",
+    "<SPECIAL_960>",
+    "<SPECIAL_961>",
+    "<SPECIAL_962>",
+    "<SPECIAL_963>",
+    "<SPECIAL_964>",
+    "<SPECIAL_965>",
+    "<SPECIAL_966>",
+    "<SPECIAL_967>",
+    "<SPECIAL_968>",
+    "<SPECIAL_969>",
+    "<SPECIAL_970>",
+    "<SPECIAL_971>",
+    "<SPECIAL_972>",
+    "<SPECIAL_973>",
+    "<SPECIAL_974>",
+    "<SPECIAL_975>",
+    "<SPECIAL_976>",
+    "<SPECIAL_977>",
+    "<SPECIAL_978>",
+    "<SPECIAL_979>",
+    "<SPECIAL_980>",
+    "<SPECIAL_981>",
+    "<SPECIAL_982>",
+    "<SPECIAL_983>",
+    "<SPECIAL_984>",
+    "<SPECIAL_985>",
+    "<SPECIAL_986>",
+    "<SPECIAL_987>",
+    "<SPECIAL_988>",
+    "<SPECIAL_989>",
+    "<SPECIAL_990>",
+    "<SPECIAL_991>",
+    "<SPECIAL_992>",
+    "<SPECIAL_993>",
+    "<SPECIAL_994>",
+    "<SPECIAL_995>",
+    "<SPECIAL_996>",
+    "<SPECIAL_997>",
+    "<SPECIAL_998>",
+    "<SPECIAL_999>"
+  ],
+  "bos_token": {
+    "content": "<s>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "eos_token": {
+    "content": "</s>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "pad_token": {
+    "content": "<pad>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "unk_token": {
+    "content": "<unk>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  }
+}

tokenizer/tokenizer.json

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

transformer/config.json

@@ -0,0 +1,22 @@
+{
+  "_class_name": "Flux2Transformer2DModel",
+  "_diffusers_version": "0.36.0.dev0",
+  "attention_head_dim": 128,
+  "axes_dims_rope": [
+    32,
+    32,
+    32,
+    32
+  ],
+  "eps": 1e-06,
+  "in_channels": 128,
+  "joint_attention_dim": 15360,
+  "mlp_ratio": 3.0,
+  "num_attention_heads": 48,
+  "num_layers": 8,
+  "num_single_layers": 48,
+  "out_channels": null,
+  "patch_size": 1,
+  "rope_theta": 2000,
+  "timestep_guidance_channels": 256
+}

transformer/diffusion_pytorch_model-00001-of-00007.safetensors

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

transformer/diffusion_pytorch_model-00002-of-00007.safetensors

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

transformer/diffusion_pytorch_model-00003-of-00007.safetensors

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

transformer/diffusion_pytorch_model-00004-of-00007.safetensors

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

transformer/diffusion_pytorch_model-00005-of-00007.safetensors

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

transformer/diffusion_pytorch_model-00006-of-00007.safetensors

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

transformer/diffusion_pytorch_model-00007-of-00007.safetensors

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

transformer/diffusion_pytorch_model.safetensors.index.json

@@ -0,0 +1,338 @@
+{
+  "metadata": {
+    "total_size": 64446562304
+  },
+  "weight_map": {
+    "context_embedder.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "double_stream_modulation_img.linear.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "double_stream_modulation_txt.linear.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "norm_out.linear.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "proj_out.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "single_stream_modulation.linear.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "single_transformer_blocks.0.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "single_transformer_blocks.0.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "single_transformer_blocks.0.attn.to_out.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "single_transformer_blocks.0.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "single_transformer_blocks.1.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "single_transformer_blocks.1.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "single_transformer_blocks.1.attn.to_out.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "single_transformer_blocks.1.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "single_transformer_blocks.10.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "single_transformer_blocks.10.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "single_transformer_blocks.10.attn.to_out.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "single_transformer_blocks.10.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "single_transformer_blocks.11.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "single_transformer_blocks.11.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "single_transformer_blocks.11.attn.to_out.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "single_transformer_blocks.11.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "single_transformer_blocks.12.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "single_transformer_blocks.12.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "single_transformer_blocks.12.attn.to_out.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "single_transformer_blocks.12.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "single_transformer_blocks.13.attn.norm_k.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "single_transformer_blocks.13.attn.norm_q.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "single_transformer_blocks.13.attn.to_out.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "single_transformer_blocks.13.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "single_transformer_blocks.14.attn.norm_k.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "single_transformer_blocks.14.attn.norm_q.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "single_transformer_blocks.14.attn.to_out.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "single_transformer_blocks.14.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "single_transformer_blocks.15.attn.norm_k.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "single_transformer_blocks.15.attn.norm_q.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "single_transformer_blocks.15.attn.to_out.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "single_transformer_blocks.15.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "single_transformer_blocks.16.attn.norm_k.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "single_transformer_blocks.16.attn.norm_q.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "single_transformer_blocks.16.attn.to_out.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "single_transformer_blocks.16.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "single_transformer_blocks.17.attn.norm_k.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "single_transformer_blocks.17.attn.norm_q.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "single_transformer_blocks.17.attn.to_out.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "single_transformer_blocks.17.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "single_transformer_blocks.18.attn.norm_k.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "single_transformer_blocks.18.attn.norm_q.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "single_transformer_blocks.18.attn.to_out.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "single_transformer_blocks.18.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "single_transformer_blocks.19.attn.norm_k.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "single_transformer_blocks.19.attn.norm_q.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "single_transformer_blocks.19.attn.to_out.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "single_transformer_blocks.19.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "single_transformer_blocks.2.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "single_transformer_blocks.2.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "single_transformer_blocks.2.attn.to_out.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "single_transformer_blocks.2.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "single_transformer_blocks.20.attn.norm_k.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "single_transformer_blocks.20.attn.norm_q.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "single_transformer_blocks.20.attn.to_out.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "single_transformer_blocks.20.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "single_transformer_blocks.21.attn.norm_k.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "single_transformer_blocks.21.attn.norm_q.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "single_transformer_blocks.21.attn.to_out.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "single_transformer_blocks.21.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "single_transformer_blocks.22.attn.norm_k.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "single_transformer_blocks.22.attn.norm_q.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "single_transformer_blocks.22.attn.to_out.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "single_transformer_blocks.22.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "single_transformer_blocks.23.attn.norm_k.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "single_transformer_blocks.23.attn.norm_q.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "single_transformer_blocks.23.attn.to_out.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "single_transformer_blocks.23.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "single_transformer_blocks.24.attn.norm_k.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "single_transformer_blocks.24.attn.norm_q.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "single_transformer_blocks.24.attn.to_out.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "single_transformer_blocks.24.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "single_transformer_blocks.25.attn.norm_k.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "single_transformer_blocks.25.attn.norm_q.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "single_transformer_blocks.25.attn.to_out.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "single_transformer_blocks.25.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "single_transformer_blocks.26.attn.norm_k.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "single_transformer_blocks.26.attn.norm_q.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "single_transformer_blocks.26.attn.to_out.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "single_transformer_blocks.26.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "single_transformer_blocks.27.attn.norm_k.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "single_transformer_blocks.27.attn.norm_q.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "single_transformer_blocks.27.attn.to_out.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "single_transformer_blocks.27.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "single_transformer_blocks.28.attn.norm_k.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "single_transformer_blocks.28.attn.norm_q.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "single_transformer_blocks.28.attn.to_out.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "single_transformer_blocks.28.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "single_transformer_blocks.29.attn.norm_k.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "single_transformer_blocks.29.attn.norm_q.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "single_transformer_blocks.29.attn.to_out.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "single_transformer_blocks.29.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "single_transformer_blocks.3.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "single_transformer_blocks.3.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "single_transformer_blocks.3.attn.to_out.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "single_transformer_blocks.3.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "single_transformer_blocks.30.attn.norm_k.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "single_transformer_blocks.30.attn.norm_q.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "single_transformer_blocks.30.attn.to_out.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "single_transformer_blocks.30.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "single_transformer_blocks.31.attn.norm_k.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "single_transformer_blocks.31.attn.norm_q.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "single_transformer_blocks.31.attn.to_out.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "single_transformer_blocks.31.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "single_transformer_blocks.32.attn.norm_k.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "single_transformer_blocks.32.attn.norm_q.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "single_transformer_blocks.32.attn.to_out.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "single_transformer_blocks.32.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "single_transformer_blocks.33.attn.norm_k.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "single_transformer_blocks.33.attn.norm_q.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "single_transformer_blocks.33.attn.to_out.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "single_transformer_blocks.33.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "single_transformer_blocks.34.attn.norm_k.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "single_transformer_blocks.34.attn.norm_q.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "single_transformer_blocks.34.attn.to_out.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "single_transformer_blocks.34.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "single_transformer_blocks.35.attn.norm_k.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "single_transformer_blocks.35.attn.norm_q.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "single_transformer_blocks.35.attn.to_out.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "single_transformer_blocks.35.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "single_transformer_blocks.36.attn.norm_k.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "single_transformer_blocks.36.attn.norm_q.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "single_transformer_blocks.36.attn.to_out.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "single_transformer_blocks.36.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "single_transformer_blocks.37.attn.norm_k.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "single_transformer_blocks.37.attn.norm_q.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "single_transformer_blocks.37.attn.to_out.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "single_transformer_blocks.37.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "single_transformer_blocks.38.attn.norm_k.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "single_transformer_blocks.38.attn.norm_q.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "single_transformer_blocks.38.attn.to_out.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "single_transformer_blocks.38.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "single_transformer_blocks.39.attn.norm_k.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "single_transformer_blocks.39.attn.norm_q.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "single_transformer_blocks.39.attn.to_out.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "single_transformer_blocks.39.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "single_transformer_blocks.4.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "single_transformer_blocks.4.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "single_transformer_blocks.4.attn.to_out.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "single_transformer_blocks.4.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "single_transformer_blocks.40.attn.norm_k.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "single_transformer_blocks.40.attn.norm_q.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "single_transformer_blocks.40.attn.to_out.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "single_transformer_blocks.40.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "single_transformer_blocks.41.attn.norm_k.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "single_transformer_blocks.41.attn.norm_q.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "single_transformer_blocks.41.attn.to_out.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "single_transformer_blocks.41.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "single_transformer_blocks.42.attn.norm_k.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "single_transformer_blocks.42.attn.norm_q.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "single_transformer_blocks.42.attn.to_out.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "single_transformer_blocks.42.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "single_transformer_blocks.43.attn.norm_k.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "single_transformer_blocks.43.attn.norm_q.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "single_transformer_blocks.43.attn.to_out.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "single_transformer_blocks.43.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "single_transformer_blocks.44.attn.norm_k.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "single_transformer_blocks.44.attn.norm_q.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "single_transformer_blocks.44.attn.to_out.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "single_transformer_blocks.44.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "single_transformer_blocks.45.attn.norm_k.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "single_transformer_blocks.45.attn.norm_q.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "single_transformer_blocks.45.attn.to_out.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "single_transformer_blocks.45.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "single_transformer_blocks.46.attn.norm_k.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "single_transformer_blocks.46.attn.norm_q.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "single_transformer_blocks.46.attn.to_out.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "single_transformer_blocks.46.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "single_transformer_blocks.47.attn.norm_k.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "single_transformer_blocks.47.attn.norm_q.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "single_transformer_blocks.47.attn.to_out.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "single_transformer_blocks.47.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "single_transformer_blocks.5.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "single_transformer_blocks.5.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "single_transformer_blocks.5.attn.to_out.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "single_transformer_blocks.5.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "single_transformer_blocks.6.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "single_transformer_blocks.6.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "single_transformer_blocks.6.attn.to_out.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "single_transformer_blocks.6.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "single_transformer_blocks.7.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "single_transformer_blocks.7.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "single_transformer_blocks.7.attn.to_out.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "single_transformer_blocks.7.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "single_transformer_blocks.8.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "single_transformer_blocks.8.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "single_transformer_blocks.8.attn.to_out.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "single_transformer_blocks.8.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "single_transformer_blocks.9.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "single_transformer_blocks.9.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "single_transformer_blocks.9.attn.to_out.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "single_transformer_blocks.9.attn.to_qkv_mlp_proj.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "time_guidance_embed.guidance_embedder.linear_1.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "time_guidance_embed.guidance_embedder.linear_2.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "time_guidance_embed.timestep_embedder.linear_1.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "time_guidance_embed.timestep_embedder.linear_2.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.0.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.0.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.0.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.0.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.0.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.0.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.0.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.0.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.0.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.0.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.0.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.0.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.0.ff.linear_in.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.0.ff.linear_out.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.0.ff_context.linear_in.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.0.ff_context.linear_out.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.1.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.1.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.1.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.1.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.1.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.1.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.1.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.1.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.1.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.1.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.1.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.1.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.1.ff.linear_in.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.1.ff.linear_out.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.1.ff_context.linear_in.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.1.ff_context.linear_out.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.2.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.2.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.2.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.2.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.2.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.2.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.2.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.2.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.2.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.2.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.2.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.2.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.2.ff.linear_in.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.2.ff.linear_out.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.2.ff_context.linear_in.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.2.ff_context.linear_out.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.3.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.3.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.3.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.3.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.3.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.3.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.3.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.3.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.3.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.3.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.3.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.3.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.3.ff.linear_in.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.3.ff.linear_out.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.3.ff_context.linear_in.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.3.ff_context.linear_out.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.4.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.4.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.4.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.4.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.4.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.4.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.4.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.4.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.4.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.4.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.4.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.4.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "transformer_blocks.4.ff.linear_in.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "transformer_blocks.4.ff.linear_out.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "transformer_blocks.4.ff_context.linear_in.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "transformer_blocks.4.ff_context.linear_out.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "transformer_blocks.5.attn.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "transformer_blocks.5.attn.add_q_proj.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "transformer_blocks.5.attn.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "transformer_blocks.5.attn.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "transformer_blocks.5.attn.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "transformer_blocks.5.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "transformer_blocks.5.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "transformer_blocks.5.attn.to_add_out.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "transformer_blocks.5.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "transformer_blocks.5.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "transformer_blocks.5.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "transformer_blocks.5.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "transformer_blocks.5.ff.linear_in.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "transformer_blocks.5.ff.linear_out.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "transformer_blocks.5.ff_context.linear_in.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "transformer_blocks.5.ff_context.linear_out.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "transformer_blocks.6.attn.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "transformer_blocks.6.attn.add_q_proj.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "transformer_blocks.6.attn.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "transformer_blocks.6.attn.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "transformer_blocks.6.attn.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "transformer_blocks.6.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "transformer_blocks.6.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "transformer_blocks.6.attn.to_add_out.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "transformer_blocks.6.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "transformer_blocks.6.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "transformer_blocks.6.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "transformer_blocks.6.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "transformer_blocks.6.ff.linear_in.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "transformer_blocks.6.ff.linear_out.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "transformer_blocks.6.ff_context.linear_in.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "transformer_blocks.6.ff_context.linear_out.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "transformer_blocks.7.attn.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "transformer_blocks.7.attn.add_q_proj.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "transformer_blocks.7.attn.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "transformer_blocks.7.attn.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "transformer_blocks.7.attn.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "transformer_blocks.7.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "transformer_blocks.7.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "transformer_blocks.7.attn.to_add_out.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "transformer_blocks.7.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "transformer_blocks.7.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "transformer_blocks.7.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "transformer_blocks.7.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "transformer_blocks.7.ff.linear_in.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "transformer_blocks.7.ff.linear_out.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "transformer_blocks.7.ff_context.linear_in.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "transformer_blocks.7.ff_context.linear_out.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "x_embedder.weight": "diffusion_pytorch_model-00001-of-00007.safetensors"
+  }
+}

vae/config.json

@@ -0,0 +1,39 @@
+{
+  "_class_name": "AutoencoderKLFlux2",
+  "_diffusers_version": "0.36.0.dev0",
+  "act_fn": "silu",
+  "batch_norm_eps": 0.0001,
+  "batch_norm_momentum": 0.1,
+  "block_out_channels": [
+    128,
+    256,
+    512,
+    512
+  ],
+  "down_block_types": [
+    "DownEncoderBlock2D",
+    "DownEncoderBlock2D",
+    "DownEncoderBlock2D",
+    "DownEncoderBlock2D"
+  ],
+  "force_upcast": true,
+  "in_channels": 3,
+  "latent_channels": 32,
+  "layers_per_block": 2,
+  "mid_block_add_attention": true,
+  "norm_num_groups": 32,
+  "out_channels": 3,
+  "patch_size": [
+    2,
+    2
+  ],
+  "sample_size": 1024,
+  "up_block_types": [
+    "UpDecoderBlock2D",
+    "UpDecoderBlock2D",
+    "UpDecoderBlock2D",
+    "UpDecoderBlock2D"
+  ],
+  "use_post_quant_conv": true,
+  "use_quant_conv": true
+}

vae/diffusion_pytorch_model.safetensors

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