Instructions to use Video-Reason/VBVR-Pro-LTX2.3 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use Video-Reason/VBVR-Pro-LTX2.3 with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline from diffusers.utils import load_image, export_to_video # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("Video-Reason/VBVR-Pro-LTX2.3", dtype=torch.bfloat16, device_map="cuda") pipe.to("cuda") prompt = "A man with short gray hair plays a red electric guitar." image = load_image( "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/guitar-man.png" ) output = pipe(image=image, prompt=prompt).frames[0] export_to_video(output, "output.mp4") - Notebooks
- Google Colab
- Kaggle
Upload folder using huggingface_hub
Browse filesThis view is limited to 50 files because it contains too many changes. See raw diff
- .gitattributes +2 -0
- README.md +217 -0
- __pycache__/example.cpython-310.pyc +0 -0
- audio_vae/config.json +25 -0
- audio_vae/diffusion_pytorch_model.safetensors +3 -0
- connectors/config.json +25 -0
- connectors/diffusion_pytorch_model-00001-of-00002.safetensors +3 -0
- connectors/diffusion_pytorch_model-00002-of-00002.safetensors +3 -0
- connectors/diffusion_pytorch_model.safetensors.index.json +269 -0
- example.py +65 -0
- model_index.json +36 -0
- processor/added_tokens.json +3 -0
- processor/chat_template.jinja +47 -0
- processor/preprocessor_config.json +29 -0
- processor/processor_config.json +4 -0
- processor/special_tokens_map.json +33 -0
- processor/tokenizer.json +3 -0
- processor/tokenizer.model +3 -0
- processor/tokenizer_config.json +0 -0
- scheduler/scheduler_config.json +18 -0
- text_encoder/config.json +114 -0
- text_encoder/generation_config.json +11 -0
- text_encoder/model-00001-of-00011.safetensors +3 -0
- text_encoder/model-00002-of-00011.safetensors +3 -0
- text_encoder/model-00003-of-00011.safetensors +3 -0
- text_encoder/model-00004-of-00011.safetensors +3 -0
- text_encoder/model-00005-of-00011.safetensors +3 -0
- text_encoder/model-00006-of-00011.safetensors +3 -0
- text_encoder/model-00007-of-00011.safetensors +3 -0
- text_encoder/model-00008-of-00011.safetensors +3 -0
- text_encoder/model-00009-of-00011.safetensors +3 -0
- text_encoder/model-00010-of-00011.safetensors +3 -0
- text_encoder/model-00011-of-00011.safetensors +3 -0
- text_encoder/model.safetensors.index.json +0 -0
- tokenizer/added_tokens.json +3 -0
- tokenizer/chat_template.jinja +47 -0
- tokenizer/special_tokens_map.json +33 -0
- tokenizer/tokenizer.json +3 -0
- tokenizer/tokenizer.model +3 -0
- tokenizer/tokenizer_config.json +0 -0
- transformer/config.json +50 -0
- transformer/diffusion_pytorch_model-00001-of-00008.safetensors +3 -0
- transformer/diffusion_pytorch_model-00002-of-00008.safetensors +3 -0
- transformer/diffusion_pytorch_model-00003-of-00008.safetensors +3 -0
- transformer/diffusion_pytorch_model-00004-of-00008.safetensors +3 -0
- transformer/diffusion_pytorch_model-00005-of-00008.safetensors +3 -0
- transformer/diffusion_pytorch_model-00006-of-00008.safetensors +3 -0
- transformer/diffusion_pytorch_model-00007-of-00008.safetensors +3 -0
- transformer/diffusion_pytorch_model-00008-of-00008.safetensors +3 -0
- transformer/diffusion_pytorch_model.safetensors.index.json +0 -0
.gitattributes
CHANGED
|
@@ -33,3 +33,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
|
| 33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
|
|
|
|
|
|
|
|
| 33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
| 36 |
+
tokenizer/tokenizer.json filter=lfs diff=lfs merge=lfs -text
|
| 37 |
+
processor/tokenizer.json filter=lfs diff=lfs merge=lfs -text
|
README.md
ADDED
|
@@ -0,0 +1,217 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
base_model:
|
| 3 |
+
- diffusers/LTX-2.3-Diffusers
|
| 4 |
+
library_name: diffusers
|
| 5 |
+
license: apache-2.0
|
| 6 |
+
pipeline_tag: image-to-video
|
| 7 |
+
tags:
|
| 8 |
+
- diffusers
|
| 9 |
+
- safetensors
|
| 10 |
+
- image-to-video
|
| 11 |
+
- text-to-video
|
| 12 |
+
- audio-video
|
| 13 |
+
- ltx-2-3
|
| 14 |
+
- vbvr
|
| 15 |
+
datasets:
|
| 16 |
+
- Video-Reason/VBVR-Dataset
|
| 17 |
+
---
|
| 18 |
+
|
| 19 |
+
# VBVR-Pro: A Scalable and Verifiable Suite for Native Visual Reasoning
|
| 20 |
+
|
| 21 |
+
<a href="https://video-reason.com/?v=pro" target="_blank">
|
| 22 |
+
<img alt="Project Page" src="https://img.shields.io/badge/Project%20-%20Homepage-4285F4" height="20" />
|
| 23 |
+
</a>
|
| 24 |
+
<a href="https://github.com/Video-Reason/VBVR-Pro-Bench" target="_blank">
|
| 25 |
+
<img alt="Code" src="https://img.shields.io/badge/Evaluation_code-VBVR_Pro_Bench-100000?style=flat-square&logo=github&logoColor=white" height="20" />
|
| 26 |
+
</a>
|
| 27 |
+
<a href="https://github.com/Video-Reason/VBVR-Pro" target="_blank">
|
| 28 |
+
<img alt="Code" src="https://img.shields.io/badge/Training_Inferenceing-VBVR_Pro-100000?style=flat-square&logo=github&logoColor=white" height="20" />
|
| 29 |
+
</a>
|
| 30 |
+
<a href="https://huggingface.co/papers/2602.20159" target="_blank">
|
| 31 |
+
<img alt="arXiv" src="https://img.shields.io/badge/arXiv-VBVR_Pro-red?logo=arxiv" height="20" />
|
| 32 |
+
</a>
|
| 33 |
+
<a href="https://huggingface.co/datasets/Video-Reason/VBVR-Pro-SFT-Video" target="_blank">
|
| 34 |
+
<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" />
|
| 35 |
+
</a>
|
| 36 |
+
<a href="https://huggingface.co/datasets/Video-Reason/VBVR-Pro-Bench/tree/main" target="_blank">
|
| 37 |
+
<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" />
|
| 38 |
+
</a>
|
| 39 |
+
<a href="https://video-reason.com/pro/bench/#leaderboard" target="_blank">
|
| 40 |
+
<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" />
|
| 41 |
+
</a>
|
| 42 |
+
|
| 43 |
+
## Overview
|
| 44 |
+
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.
|
| 45 |
+
|
| 46 |
+
The models are presented in the paper [VBVR-Pro: A Scalable and Verifiable Suite for Native Visual Reasoning](https://huggingface.co/papers/2602.20159).
|
| 47 |
+
|
| 48 |
+
## Models Zoo
|
| 49 |
+
<table border="1" cellspacing="0" cellpadding="4" style="border-collapse: collapse; width: 100%;">
|
| 50 |
+
<thead>
|
| 51 |
+
<tr>
|
| 52 |
+
<th width="260" style="min-width: 260px;">Model</th>
|
| 53 |
+
<th>Base Architecture</th>
|
| 54 |
+
<th>Other Remarks</th>
|
| 55 |
+
</tr>
|
| 56 |
+
</thead>
|
| 57 |
+
<tbody>
|
| 58 |
+
<tr style="border-top: 4px solid #6b7280; background-color: #e5e7eb;"><th colspan="3" align="left">Image Generation Models</th></tr>
|
| 59 |
+
<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>
|
| 60 |
+
<tr><td><a href="https://huggingface.co/Video-Reason/VBVR-Pro-FLUX2-dev">VBVR-Pro-FLUX2-dev</a></td><td>FLUX.2-dev</td><td>Complete model, Diffusers format</td></tr>
|
| 61 |
+
<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>
|
| 62 |
+
<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>
|
| 63 |
+
<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>
|
| 64 |
+
<tr style="border-top: 4px solid #6b7280; background-color: #e5e7eb;"><th colspan="3" align="left">Interleaved Image Generation Models</th></tr>
|
| 65 |
+
<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>
|
| 66 |
+
<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>
|
| 67 |
+
<tr style="border-top: 4px solid #6b7280; background-color: #e5e7eb;"><th colspan="3" align="left">Video Generation Models</th></tr>
|
| 68 |
+
<tr><td><a href="https://huggingface.co/Video-Reason/VBVR-Pro-LTX2.3"><strong>VBVR-Pro-LTX2.3</strong></a></td><td>LTX-Video-2.3</td><td>Complete model, Diffusers format</td></tr>
|
| 69 |
+
<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>
|
| 70 |
+
<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>
|
| 71 |
+
<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>
|
| 72 |
+
<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>
|
| 73 |
+
<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>
|
| 74 |
+
<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>
|
| 75 |
+
<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>
|
| 76 |
+
</tbody>
|
| 77 |
+
</table>
|
| 78 |
+
|
| 79 |
+
## Release Information
|
| 80 |
+
|
| 81 |
+
VBVR-Pro fine-tuned **LTX-2.3** for instruction-based, image-conditioned
|
| 82 |
+
audio-video generation. It was trained on the
|
| 83 |
+
[VBVR-Dataset](https://huggingface.co/datasets/Video-Reason/VBVR-Dataset).
|
| 84 |
+
|
| 85 |
+
This is the **merged model**: the rank-32 LoRA has already been fused into the
|
| 86 |
+
LTX transformer weights. Load it directly with
|
| 87 |
+
[diffusers](https://github.com/huggingface/diffusers); no adapter needs to be
|
| 88 |
+
loaded separately.
|
| 89 |
+
|
| 90 |
+
| Format | Repository |
|
| 91 |
+
|---|---|
|
| 92 |
+
| Merged / Diffusers (this repo) | `VBVR-Pro-LTX2.3` |
|
| 93 |
+
| DiffSynth LoRA | `VBVR-Pro-LTX2.3-diffsynth` |
|
| 94 |
+
|
| 95 |
+
In this release, we present all models presented in paper
|
| 96 |
+
[**VBVR-Pro-Trained-Models**](https://huggingface.co/collections/Video-Reason/VBVR-Pro),
|
| 97 |
+
[**VBVR-Pro-Dataset-Video**](https://huggingface.co/datasets/Video-Reason/VBVR-Pro-SFT-Video),
|
| 98 |
+
[**VBVR-Pro-Dataset-Image**](https://huggingface.co/datasets/Video-Reason/VBVR-Pro-SFT-Image),
|
| 99 |
+
[**VBVR-Pro-Bench**](https://huggingface.co/datasets/Video-Reason/VBVR-Pro-Bench),
|
| 100 |
+
[**VBVR-Pro-Code**](https://github.com/Video-Reason/VBVR-Pro) and
|
| 101 |
+
[**VBVR-Bench-Leaderboard**](https://video-reason.com/pro/bench/#leaderboard).
|
| 102 |
+
|
| 103 |
+
## VBVR-Pro Benchmark Results
|
| 104 |
+
<table border="1" cellspacing="0" cellpadding="4" style="border-collapse: collapse; width: 100%; font-size: 12px;">
|
| 105 |
+
<thead>
|
| 106 |
+
<tr>
|
| 107 |
+
<th rowspan="2" width="260" style="min-width: 260px;">Models</th>
|
| 108 |
+
<th rowspan="2">Overall</th>
|
| 109 |
+
<th colspan="6">In-Domain by Category</th>
|
| 110 |
+
<th colspan="6">Out-of-Domain by Category</th>
|
| 111 |
+
</tr>
|
| 112 |
+
<tr>
|
| 113 |
+
<th>Avg.</th><th>Abst.</th><th>Know.</th><th>Perc.</th><th>Spat.</th><th>Trans.</th>
|
| 114 |
+
<th>Avg.</th><th>Abst.</th><th>Know.</th><th>Perc.</th><th>Spat.</th><th>Trans.</th>
|
| 115 |
+
</tr>
|
| 116 |
+
</thead>
|
| 117 |
+
<tbody>
|
| 118 |
+
<tr style="border-top: 4px solid #6b7280; background-color: #e5e7eb;"><th colspan="14" align="left">Image Generation Models</th></tr>
|
| 119 |
+
<tr style="background-color: #dbeafe;"><th colspan="14" align="left">Proprietary Models</th></tr>
|
| 120 |
+
<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>
|
| 121 |
+
<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>
|
| 122 |
+
<tr style="background-color: #dbeafe;"><th colspan="14" align="left">Open-source Models</th></tr>
|
| 123 |
+
<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>
|
| 124 |
+
<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>
|
| 125 |
+
<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>
|
| 126 |
+
<tr style="background-color: #dbeafe;"><th colspan="14" align="left">Strong Baselines</th></tr>
|
| 127 |
+
<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>
|
| 128 |
+
<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>
|
| 129 |
+
<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>
|
| 130 |
+
<tr style="border-top: 4px solid #6b7280; background-color: #e5e7eb;"><th colspan="14" align="left">Interleaved Image Generation Models</th></tr>
|
| 131 |
+
<tr style="background-color: #dbeafe;"><th colspan="14" align="left">Proprietary Models</th></tr>
|
| 132 |
+
<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>
|
| 133 |
+
<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>
|
| 134 |
+
<tr style="background-color: #dbeafe;"><th colspan="14" align="left">Open-source Models</th></tr>
|
| 135 |
+
<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>
|
| 136 |
+
<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>
|
| 137 |
+
<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>
|
| 138 |
+
<tr style="background-color: #dbeafe;"><th colspan="14" align="left">Strong Baselines</th></tr>
|
| 139 |
+
<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>
|
| 140 |
+
<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>
|
| 141 |
+
<tr style="border-top: 4px solid #6b7280; background-color: #e5e7eb;"><th colspan="14" align="left">Video Generation Models</th></tr>
|
| 142 |
+
<tr style="background-color: #dbeafe;"><th colspan="14" align="left">Proprietary Models</th></tr>
|
| 143 |
+
<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>
|
| 144 |
+
<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>
|
| 145 |
+
<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>
|
| 146 |
+
<tr style="background-color: #dbeafe;"><th colspan="14" align="left">Open-source Models</th></tr>
|
| 147 |
+
<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>
|
| 148 |
+
<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>
|
| 149 |
+
<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>
|
| 150 |
+
<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>
|
| 151 |
+
<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>
|
| 152 |
+
<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>
|
| 153 |
+
<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>
|
| 154 |
+
<tr style="background-color: #dbeafe;"><th colspan="14" align="left">Strong Baselines</th></tr>
|
| 155 |
+
<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>
|
| 156 |
+
<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>
|
| 157 |
+
<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>
|
| 158 |
+
<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>
|
| 159 |
+
</tbody>
|
| 160 |
+
</table>
|
| 161 |
+
|
| 162 |
+
## Quick Start
|
| 163 |
+
|
| 164 |
+
### Method 1: Standalone Diffusers inference
|
| 165 |
+
|
| 166 |
+
#### 1. Install Diffusers
|
| 167 |
+
|
| 168 |
+
```bash
|
| 169 |
+
pip install -U diffusers transformers accelerate pillow imageio imageio-ffmpeg
|
| 170 |
+
```
|
| 171 |
+
|
| 172 |
+
#### 2. Run `example.py`
|
| 173 |
+
|
| 174 |
+
The included [`example.py`](example.py) loads the merged checkpoint directly
|
| 175 |
+
with Diffusers, enables model CPU offloading, and writes synchronized audio and
|
| 176 |
+
video to an MP4 file.
|
| 177 |
+
|
| 178 |
+
```bash
|
| 179 |
+
python example.py \
|
| 180 |
+
--model_path Video-Reason/VBVR-Pro-LTX2.3 \
|
| 181 |
+
--image input.png \
|
| 182 |
+
--prompt "The machine starts and makes a quiet mechanical hum." \
|
| 183 |
+
--num_frames 49 --fps 24 \
|
| 184 |
+
--output output.mp4
|
| 185 |
+
```
|
| 186 |
+
|
| 187 |
+
### Method 2: Unified VBVR-Pro inference
|
| 188 |
+
|
| 189 |
+
Clone [Video-Reason/VBVR-Pro](https://github.com/Video-Reason/VBVR-Pro) and
|
| 190 |
+
create its unified inference environment:
|
| 191 |
+
|
| 192 |
+
```bash
|
| 193 |
+
git clone https://github.com/Video-Reason/VBVR-Pro.git
|
| 194 |
+
cd VBVR-Pro/
|
| 195 |
+
uv sync --extra cu124 # or one of [cu118|cu121|cu124|cu126|cu128|cu129]
|
| 196 |
+
source .venv/bin/activate
|
| 197 |
+
```
|
| 198 |
+
|
| 199 |
+
Then run the unified inference script:
|
| 200 |
+
|
| 201 |
+
```bash
|
| 202 |
+
python example.py \
|
| 203 |
+
--model_path Video-Reason/VBVR-Pro-LTX2.3 \
|
| 204 |
+
--image_paths input.png \
|
| 205 |
+
--prompt "The machine starts and makes a quiet mechanical hum." \
|
| 206 |
+
--num_frames 49 --fps 24 \
|
| 207 |
+
--output output.mp4
|
| 208 |
+
```
|
| 209 |
+
|
| 210 |
+
## Citation
|
| 211 |
+
|
| 212 |
+
```bibtex
|
| 213 |
+
@article{vbvr2025,
|
| 214 |
+
title={VBVR: A Very Big Video Reasoning Suite},
|
| 215 |
+
year={2025},
|
| 216 |
+
}
|
| 217 |
+
```
|
__pycache__/example.cpython-310.pyc
ADDED
|
Binary file (1.97 kB). View file
|
|
|
audio_vae/config.json
ADDED
|
@@ -0,0 +1,25 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"_class_name": "AutoencoderKLLTX2Audio",
|
| 3 |
+
"_diffusers_version": "0.37.0.dev0",
|
| 4 |
+
"attn_resolutions": null,
|
| 5 |
+
"base_channels": 128,
|
| 6 |
+
"causality_axis": "height",
|
| 7 |
+
"ch_mult": [
|
| 8 |
+
1,
|
| 9 |
+
2,
|
| 10 |
+
4
|
| 11 |
+
],
|
| 12 |
+
"double_z": true,
|
| 13 |
+
"dropout": 0.0,
|
| 14 |
+
"in_channels": 2,
|
| 15 |
+
"is_causal": true,
|
| 16 |
+
"latent_channels": 8,
|
| 17 |
+
"mel_bins": 64,
|
| 18 |
+
"mel_hop_length": 160,
|
| 19 |
+
"mid_block_add_attention": false,
|
| 20 |
+
"norm_type": "pixel",
|
| 21 |
+
"num_res_blocks": 2,
|
| 22 |
+
"output_channels": 2,
|
| 23 |
+
"resolution": 256,
|
| 24 |
+
"sample_rate": 16000
|
| 25 |
+
}
|
audio_vae/diffusion_pytorch_model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:45dcb1c3cffc8c8b7710a04dd19500e481daff31f5ea8e8529f2cf697c439d12
|
| 3 |
+
size 106507972
|
connectors/config.json
ADDED
|
@@ -0,0 +1,25 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"_class_name": "LTX2TextConnectors",
|
| 3 |
+
"_diffusers_version": "0.37.0.dev0",
|
| 4 |
+
"audio_connector_attention_head_dim": 64,
|
| 5 |
+
"audio_connector_num_attention_heads": 32,
|
| 6 |
+
"audio_connector_num_layers": 8,
|
| 7 |
+
"audio_connector_num_learnable_registers": 128,
|
| 8 |
+
"audio_gated_attn": true,
|
| 9 |
+
"audio_hidden_dim": 2048,
|
| 10 |
+
"caption_channels": 3840,
|
| 11 |
+
"causal_temporal_positioning": false,
|
| 12 |
+
"connector_rope_base_seq_len": 4096,
|
| 13 |
+
"per_modality_projections": true,
|
| 14 |
+
"proj_bias": true,
|
| 15 |
+
"rope_double_precision": true,
|
| 16 |
+
"rope_theta": 10000.0,
|
| 17 |
+
"rope_type": "split",
|
| 18 |
+
"text_proj_in_factor": 49,
|
| 19 |
+
"video_connector_attention_head_dim": 128,
|
| 20 |
+
"video_connector_num_attention_heads": 32,
|
| 21 |
+
"video_connector_num_layers": 8,
|
| 22 |
+
"video_connector_num_learnable_registers": 128,
|
| 23 |
+
"video_gated_attn": true,
|
| 24 |
+
"video_hidden_dim": 4096
|
| 25 |
+
}
|
connectors/diffusion_pytorch_model-00001-of-00002.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:b767a411e336d692306d7d34fa85627db1d17809a4690f45dd96a4b3807d248c
|
| 3 |
+
size 4999998520
|
connectors/diffusion_pytorch_model-00002-of-00002.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:ffe33521b2bbd12604abe2d63b5cf02cf55f2fdf10226ff5aae00e437678ebc8
|
| 3 |
+
size 1344490576
|
connectors/diffusion_pytorch_model.safetensors.index.json
ADDED
|
@@ -0,0 +1,269 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"metadata": {
|
| 3 |
+
"total_size": 6344455168
|
| 4 |
+
},
|
| 5 |
+
"weight_map": {
|
| 6 |
+
"audio_connector.learnable_registers": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 7 |
+
"audio_connector.transformer_blocks.0.attn1.norm_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 8 |
+
"audio_connector.transformer_blocks.0.attn1.norm_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 9 |
+
"audio_connector.transformer_blocks.0.attn1.to_gate_logits.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 10 |
+
"audio_connector.transformer_blocks.0.attn1.to_gate_logits.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 11 |
+
"audio_connector.transformer_blocks.0.attn1.to_k.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 12 |
+
"audio_connector.transformer_blocks.0.attn1.to_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 13 |
+
"audio_connector.transformer_blocks.0.attn1.to_out.0.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 14 |
+
"audio_connector.transformer_blocks.0.attn1.to_out.0.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 15 |
+
"audio_connector.transformer_blocks.0.attn1.to_q.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 16 |
+
"audio_connector.transformer_blocks.0.attn1.to_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 17 |
+
"audio_connector.transformer_blocks.0.attn1.to_v.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 18 |
+
"audio_connector.transformer_blocks.0.attn1.to_v.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 19 |
+
"audio_connector.transformer_blocks.0.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 20 |
+
"audio_connector.transformer_blocks.0.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 21 |
+
"audio_connector.transformer_blocks.0.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 22 |
+
"audio_connector.transformer_blocks.0.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 23 |
+
"audio_connector.transformer_blocks.1.attn1.norm_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 24 |
+
"audio_connector.transformer_blocks.1.attn1.norm_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 25 |
+
"audio_connector.transformer_blocks.1.attn1.to_gate_logits.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 26 |
+
"audio_connector.transformer_blocks.1.attn1.to_gate_logits.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 27 |
+
"audio_connector.transformer_blocks.1.attn1.to_k.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 28 |
+
"audio_connector.transformer_blocks.1.attn1.to_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 29 |
+
"audio_connector.transformer_blocks.1.attn1.to_out.0.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 30 |
+
"audio_connector.transformer_blocks.1.attn1.to_out.0.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 31 |
+
"audio_connector.transformer_blocks.1.attn1.to_q.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 32 |
+
"audio_connector.transformer_blocks.1.attn1.to_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 33 |
+
"audio_connector.transformer_blocks.1.attn1.to_v.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 34 |
+
"audio_connector.transformer_blocks.1.attn1.to_v.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 35 |
+
"audio_connector.transformer_blocks.1.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 36 |
+
"audio_connector.transformer_blocks.1.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 37 |
+
"audio_connector.transformer_blocks.1.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 38 |
+
"audio_connector.transformer_blocks.1.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 39 |
+
"audio_connector.transformer_blocks.2.attn1.norm_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 40 |
+
"audio_connector.transformer_blocks.2.attn1.norm_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 41 |
+
"audio_connector.transformer_blocks.2.attn1.to_gate_logits.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 42 |
+
"audio_connector.transformer_blocks.2.attn1.to_gate_logits.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 43 |
+
"audio_connector.transformer_blocks.2.attn1.to_k.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 44 |
+
"audio_connector.transformer_blocks.2.attn1.to_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 45 |
+
"audio_connector.transformer_blocks.2.attn1.to_out.0.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 46 |
+
"audio_connector.transformer_blocks.2.attn1.to_out.0.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 47 |
+
"audio_connector.transformer_blocks.2.attn1.to_q.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 48 |
+
"audio_connector.transformer_blocks.2.attn1.to_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 49 |
+
"audio_connector.transformer_blocks.2.attn1.to_v.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 50 |
+
"audio_connector.transformer_blocks.2.attn1.to_v.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 51 |
+
"audio_connector.transformer_blocks.2.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 52 |
+
"audio_connector.transformer_blocks.2.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 53 |
+
"audio_connector.transformer_blocks.2.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 54 |
+
"audio_connector.transformer_blocks.2.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 55 |
+
"audio_connector.transformer_blocks.3.attn1.norm_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 56 |
+
"audio_connector.transformer_blocks.3.attn1.norm_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 57 |
+
"audio_connector.transformer_blocks.3.attn1.to_gate_logits.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 58 |
+
"audio_connector.transformer_blocks.3.attn1.to_gate_logits.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 59 |
+
"audio_connector.transformer_blocks.3.attn1.to_k.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 60 |
+
"audio_connector.transformer_blocks.3.attn1.to_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 61 |
+
"audio_connector.transformer_blocks.3.attn1.to_out.0.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 62 |
+
"audio_connector.transformer_blocks.3.attn1.to_out.0.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 63 |
+
"audio_connector.transformer_blocks.3.attn1.to_q.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 64 |
+
"audio_connector.transformer_blocks.3.attn1.to_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 65 |
+
"audio_connector.transformer_blocks.3.attn1.to_v.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 66 |
+
"audio_connector.transformer_blocks.3.attn1.to_v.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 67 |
+
"audio_connector.transformer_blocks.3.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 68 |
+
"audio_connector.transformer_blocks.3.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 69 |
+
"audio_connector.transformer_blocks.3.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 70 |
+
"audio_connector.transformer_blocks.3.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 71 |
+
"audio_connector.transformer_blocks.4.attn1.norm_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 72 |
+
"audio_connector.transformer_blocks.4.attn1.norm_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 73 |
+
"audio_connector.transformer_blocks.4.attn1.to_gate_logits.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 74 |
+
"audio_connector.transformer_blocks.4.attn1.to_gate_logits.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 75 |
+
"audio_connector.transformer_blocks.4.attn1.to_k.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 76 |
+
"audio_connector.transformer_blocks.4.attn1.to_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 77 |
+
"audio_connector.transformer_blocks.4.attn1.to_out.0.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 78 |
+
"audio_connector.transformer_blocks.4.attn1.to_out.0.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 79 |
+
"audio_connector.transformer_blocks.4.attn1.to_q.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 80 |
+
"audio_connector.transformer_blocks.4.attn1.to_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 81 |
+
"audio_connector.transformer_blocks.4.attn1.to_v.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 82 |
+
"audio_connector.transformer_blocks.4.attn1.to_v.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 83 |
+
"audio_connector.transformer_blocks.4.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 84 |
+
"audio_connector.transformer_blocks.4.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 85 |
+
"audio_connector.transformer_blocks.4.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 86 |
+
"audio_connector.transformer_blocks.4.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 87 |
+
"audio_connector.transformer_blocks.5.attn1.norm_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 88 |
+
"audio_connector.transformer_blocks.5.attn1.norm_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 89 |
+
"audio_connector.transformer_blocks.5.attn1.to_gate_logits.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 90 |
+
"audio_connector.transformer_blocks.5.attn1.to_gate_logits.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 91 |
+
"audio_connector.transformer_blocks.5.attn1.to_k.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 92 |
+
"audio_connector.transformer_blocks.5.attn1.to_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 93 |
+
"audio_connector.transformer_blocks.5.attn1.to_out.0.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 94 |
+
"audio_connector.transformer_blocks.5.attn1.to_out.0.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 95 |
+
"audio_connector.transformer_blocks.5.attn1.to_q.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 96 |
+
"audio_connector.transformer_blocks.5.attn1.to_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 97 |
+
"audio_connector.transformer_blocks.5.attn1.to_v.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 98 |
+
"audio_connector.transformer_blocks.5.attn1.to_v.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 99 |
+
"audio_connector.transformer_blocks.5.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 100 |
+
"audio_connector.transformer_blocks.5.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 101 |
+
"audio_connector.transformer_blocks.5.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 102 |
+
"audio_connector.transformer_blocks.5.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 103 |
+
"audio_connector.transformer_blocks.6.attn1.norm_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 104 |
+
"audio_connector.transformer_blocks.6.attn1.norm_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 105 |
+
"audio_connector.transformer_blocks.6.attn1.to_gate_logits.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 106 |
+
"audio_connector.transformer_blocks.6.attn1.to_gate_logits.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 107 |
+
"audio_connector.transformer_blocks.6.attn1.to_k.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 108 |
+
"audio_connector.transformer_blocks.6.attn1.to_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 109 |
+
"audio_connector.transformer_blocks.6.attn1.to_out.0.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 110 |
+
"audio_connector.transformer_blocks.6.attn1.to_out.0.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 111 |
+
"audio_connector.transformer_blocks.6.attn1.to_q.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 112 |
+
"audio_connector.transformer_blocks.6.attn1.to_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 113 |
+
"audio_connector.transformer_blocks.6.attn1.to_v.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 114 |
+
"audio_connector.transformer_blocks.6.attn1.to_v.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 115 |
+
"audio_connector.transformer_blocks.6.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 116 |
+
"audio_connector.transformer_blocks.6.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 117 |
+
"audio_connector.transformer_blocks.6.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 118 |
+
"audio_connector.transformer_blocks.6.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 119 |
+
"audio_connector.transformer_blocks.7.attn1.norm_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 120 |
+
"audio_connector.transformer_blocks.7.attn1.norm_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 121 |
+
"audio_connector.transformer_blocks.7.attn1.to_gate_logits.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 122 |
+
"audio_connector.transformer_blocks.7.attn1.to_gate_logits.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 123 |
+
"audio_connector.transformer_blocks.7.attn1.to_k.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 124 |
+
"audio_connector.transformer_blocks.7.attn1.to_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 125 |
+
"audio_connector.transformer_blocks.7.attn1.to_out.0.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 126 |
+
"audio_connector.transformer_blocks.7.attn1.to_out.0.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 127 |
+
"audio_connector.transformer_blocks.7.attn1.to_q.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 128 |
+
"audio_connector.transformer_blocks.7.attn1.to_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 129 |
+
"audio_connector.transformer_blocks.7.attn1.to_v.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 130 |
+
"audio_connector.transformer_blocks.7.attn1.to_v.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 131 |
+
"audio_connector.transformer_blocks.7.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 132 |
+
"audio_connector.transformer_blocks.7.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 133 |
+
"audio_connector.transformer_blocks.7.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 134 |
+
"audio_connector.transformer_blocks.7.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 135 |
+
"audio_text_proj_in.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 136 |
+
"audio_text_proj_in.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 137 |
+
"video_connector.learnable_registers": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 138 |
+
"video_connector.transformer_blocks.0.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 139 |
+
"video_connector.transformer_blocks.0.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 140 |
+
"video_connector.transformer_blocks.0.attn1.to_gate_logits.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 141 |
+
"video_connector.transformer_blocks.0.attn1.to_gate_logits.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 142 |
+
"video_connector.transformer_blocks.0.attn1.to_k.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 143 |
+
"video_connector.transformer_blocks.0.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 144 |
+
"video_connector.transformer_blocks.0.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 145 |
+
"video_connector.transformer_blocks.0.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 146 |
+
"video_connector.transformer_blocks.0.attn1.to_q.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 147 |
+
"video_connector.transformer_blocks.0.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 148 |
+
"video_connector.transformer_blocks.0.attn1.to_v.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 149 |
+
"video_connector.transformer_blocks.0.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 150 |
+
"video_connector.transformer_blocks.0.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 151 |
+
"video_connector.transformer_blocks.0.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 152 |
+
"video_connector.transformer_blocks.0.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 153 |
+
"video_connector.transformer_blocks.0.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 154 |
+
"video_connector.transformer_blocks.1.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 155 |
+
"video_connector.transformer_blocks.1.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 156 |
+
"video_connector.transformer_blocks.1.attn1.to_gate_logits.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 157 |
+
"video_connector.transformer_blocks.1.attn1.to_gate_logits.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 158 |
+
"video_connector.transformer_blocks.1.attn1.to_k.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 159 |
+
"video_connector.transformer_blocks.1.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 160 |
+
"video_connector.transformer_blocks.1.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 161 |
+
"video_connector.transformer_blocks.1.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 162 |
+
"video_connector.transformer_blocks.1.attn1.to_q.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 163 |
+
"video_connector.transformer_blocks.1.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 164 |
+
"video_connector.transformer_blocks.1.attn1.to_v.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 165 |
+
"video_connector.transformer_blocks.1.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 166 |
+
"video_connector.transformer_blocks.1.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 167 |
+
"video_connector.transformer_blocks.1.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 168 |
+
"video_connector.transformer_blocks.1.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 169 |
+
"video_connector.transformer_blocks.1.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 170 |
+
"video_connector.transformer_blocks.2.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 171 |
+
"video_connector.transformer_blocks.2.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 172 |
+
"video_connector.transformer_blocks.2.attn1.to_gate_logits.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 173 |
+
"video_connector.transformer_blocks.2.attn1.to_gate_logits.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 174 |
+
"video_connector.transformer_blocks.2.attn1.to_k.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 175 |
+
"video_connector.transformer_blocks.2.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 176 |
+
"video_connector.transformer_blocks.2.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 177 |
+
"video_connector.transformer_blocks.2.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 178 |
+
"video_connector.transformer_blocks.2.attn1.to_q.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 179 |
+
"video_connector.transformer_blocks.2.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 180 |
+
"video_connector.transformer_blocks.2.attn1.to_v.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 181 |
+
"video_connector.transformer_blocks.2.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 182 |
+
"video_connector.transformer_blocks.2.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 183 |
+
"video_connector.transformer_blocks.2.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 184 |
+
"video_connector.transformer_blocks.2.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 185 |
+
"video_connector.transformer_blocks.2.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 186 |
+
"video_connector.transformer_blocks.3.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 187 |
+
"video_connector.transformer_blocks.3.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 188 |
+
"video_connector.transformer_blocks.3.attn1.to_gate_logits.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 189 |
+
"video_connector.transformer_blocks.3.attn1.to_gate_logits.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 190 |
+
"video_connector.transformer_blocks.3.attn1.to_k.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 191 |
+
"video_connector.transformer_blocks.3.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 192 |
+
"video_connector.transformer_blocks.3.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 193 |
+
"video_connector.transformer_blocks.3.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 194 |
+
"video_connector.transformer_blocks.3.attn1.to_q.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 195 |
+
"video_connector.transformer_blocks.3.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 196 |
+
"video_connector.transformer_blocks.3.attn1.to_v.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 197 |
+
"video_connector.transformer_blocks.3.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 198 |
+
"video_connector.transformer_blocks.3.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 199 |
+
"video_connector.transformer_blocks.3.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 200 |
+
"video_connector.transformer_blocks.3.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 201 |
+
"video_connector.transformer_blocks.3.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 202 |
+
"video_connector.transformer_blocks.4.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 203 |
+
"video_connector.transformer_blocks.4.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 204 |
+
"video_connector.transformer_blocks.4.attn1.to_gate_logits.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 205 |
+
"video_connector.transformer_blocks.4.attn1.to_gate_logits.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 206 |
+
"video_connector.transformer_blocks.4.attn1.to_k.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 207 |
+
"video_connector.transformer_blocks.4.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 208 |
+
"video_connector.transformer_blocks.4.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 209 |
+
"video_connector.transformer_blocks.4.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 210 |
+
"video_connector.transformer_blocks.4.attn1.to_q.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 211 |
+
"video_connector.transformer_blocks.4.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 212 |
+
"video_connector.transformer_blocks.4.attn1.to_v.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 213 |
+
"video_connector.transformer_blocks.4.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 214 |
+
"video_connector.transformer_blocks.4.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 215 |
+
"video_connector.transformer_blocks.4.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 216 |
+
"video_connector.transformer_blocks.4.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 217 |
+
"video_connector.transformer_blocks.4.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 218 |
+
"video_connector.transformer_blocks.5.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 219 |
+
"video_connector.transformer_blocks.5.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 220 |
+
"video_connector.transformer_blocks.5.attn1.to_gate_logits.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 221 |
+
"video_connector.transformer_blocks.5.attn1.to_gate_logits.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 222 |
+
"video_connector.transformer_blocks.5.attn1.to_k.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 223 |
+
"video_connector.transformer_blocks.5.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 224 |
+
"video_connector.transformer_blocks.5.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 225 |
+
"video_connector.transformer_blocks.5.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 226 |
+
"video_connector.transformer_blocks.5.attn1.to_q.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 227 |
+
"video_connector.transformer_blocks.5.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 228 |
+
"video_connector.transformer_blocks.5.attn1.to_v.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 229 |
+
"video_connector.transformer_blocks.5.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 230 |
+
"video_connector.transformer_blocks.5.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 231 |
+
"video_connector.transformer_blocks.5.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 232 |
+
"video_connector.transformer_blocks.5.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 233 |
+
"video_connector.transformer_blocks.5.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 234 |
+
"video_connector.transformer_blocks.6.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 235 |
+
"video_connector.transformer_blocks.6.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 236 |
+
"video_connector.transformer_blocks.6.attn1.to_gate_logits.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 237 |
+
"video_connector.transformer_blocks.6.attn1.to_gate_logits.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 238 |
+
"video_connector.transformer_blocks.6.attn1.to_k.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 239 |
+
"video_connector.transformer_blocks.6.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 240 |
+
"video_connector.transformer_blocks.6.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 241 |
+
"video_connector.transformer_blocks.6.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 242 |
+
"video_connector.transformer_blocks.6.attn1.to_q.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 243 |
+
"video_connector.transformer_blocks.6.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 244 |
+
"video_connector.transformer_blocks.6.attn1.to_v.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 245 |
+
"video_connector.transformer_blocks.6.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 246 |
+
"video_connector.transformer_blocks.6.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 247 |
+
"video_connector.transformer_blocks.6.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 248 |
+
"video_connector.transformer_blocks.6.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 249 |
+
"video_connector.transformer_blocks.6.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 250 |
+
"video_connector.transformer_blocks.7.attn1.norm_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 251 |
+
"video_connector.transformer_blocks.7.attn1.norm_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 252 |
+
"video_connector.transformer_blocks.7.attn1.to_gate_logits.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 253 |
+
"video_connector.transformer_blocks.7.attn1.to_gate_logits.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 254 |
+
"video_connector.transformer_blocks.7.attn1.to_k.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 255 |
+
"video_connector.transformer_blocks.7.attn1.to_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 256 |
+
"video_connector.transformer_blocks.7.attn1.to_out.0.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 257 |
+
"video_connector.transformer_blocks.7.attn1.to_out.0.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 258 |
+
"video_connector.transformer_blocks.7.attn1.to_q.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 259 |
+
"video_connector.transformer_blocks.7.attn1.to_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 260 |
+
"video_connector.transformer_blocks.7.attn1.to_v.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 261 |
+
"video_connector.transformer_blocks.7.attn1.to_v.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 262 |
+
"video_connector.transformer_blocks.7.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 263 |
+
"video_connector.transformer_blocks.7.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 264 |
+
"video_connector.transformer_blocks.7.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 265 |
+
"video_connector.transformer_blocks.7.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
|
| 266 |
+
"video_text_proj_in.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
|
| 267 |
+
"video_text_proj_in.weight": "diffusion_pytorch_model-00001-of-00002.safetensors"
|
| 268 |
+
}
|
| 269 |
+
}
|
example.py
ADDED
|
@@ -0,0 +1,65 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
"""VBVR-Pro-LTX2.3 image-to-audio-video inference example.
|
| 2 |
+
|
| 3 |
+
Usage:
|
| 4 |
+
python example.py --model_path Video-Reason/VBVR-Pro-LTX2.3 \
|
| 5 |
+
--image input.png --prompt "Your video instruction"
|
| 6 |
+
"""
|
| 7 |
+
|
| 8 |
+
import argparse
|
| 9 |
+
|
| 10 |
+
import torch
|
| 11 |
+
from diffusers import LTX2ImageToVideoPipeline
|
| 12 |
+
from diffusers.utils import encode_video, load_image
|
| 13 |
+
|
| 14 |
+
|
| 15 |
+
parser = argparse.ArgumentParser()
|
| 16 |
+
parser.add_argument("--model_path", type=str, default="Video-Reason/VBVR-Pro-LTX2.3")
|
| 17 |
+
parser.add_argument("--image", type=str, required=True, help="Path or URL to input image")
|
| 18 |
+
parser.add_argument("--prompt", type=str, required=True, help="Video instruction")
|
| 19 |
+
parser.add_argument(
|
| 20 |
+
"--negative_prompt",
|
| 21 |
+
type=str,
|
| 22 |
+
default="blurry, low quality, flickering, motion blur, distorted",
|
| 23 |
+
)
|
| 24 |
+
parser.add_argument("--output", type=str, default="output.mp4")
|
| 25 |
+
parser.add_argument("--width", type=int, default=768)
|
| 26 |
+
parser.add_argument("--height", type=int, default=512)
|
| 27 |
+
parser.add_argument("--num_frames", type=int, default=49)
|
| 28 |
+
parser.add_argument("--steps", type=int, default=40)
|
| 29 |
+
parser.add_argument("--guidance_scale", type=float, default=5.0)
|
| 30 |
+
parser.add_argument("--fps", type=int, default=24)
|
| 31 |
+
parser.add_argument("--seed", type=int, default=42)
|
| 32 |
+
args = parser.parse_args()
|
| 33 |
+
|
| 34 |
+
print(f"Loading model from: {args.model_path}")
|
| 35 |
+
pipe = LTX2ImageToVideoPipeline.from_pretrained(
|
| 36 |
+
args.model_path, torch_dtype=torch.bfloat16
|
| 37 |
+
)
|
| 38 |
+
pipe.enable_model_cpu_offload()
|
| 39 |
+
|
| 40 |
+
image = load_image(args.image).convert("RGB")
|
| 41 |
+
print(f"Input image: {args.image} ({image.size[0]}x{image.size[1]})")
|
| 42 |
+
|
| 43 |
+
video, audio = pipe(
|
| 44 |
+
image=image,
|
| 45 |
+
prompt=args.prompt,
|
| 46 |
+
negative_prompt=args.negative_prompt,
|
| 47 |
+
height=args.height,
|
| 48 |
+
width=args.width,
|
| 49 |
+
num_frames=args.num_frames,
|
| 50 |
+
frame_rate=args.fps,
|
| 51 |
+
num_inference_steps=args.steps,
|
| 52 |
+
guidance_scale=args.guidance_scale,
|
| 53 |
+
generator=torch.manual_seed(args.seed),
|
| 54 |
+
output_type="np",
|
| 55 |
+
return_dict=False,
|
| 56 |
+
)
|
| 57 |
+
|
| 58 |
+
encode_video(
|
| 59 |
+
video[0][: args.num_frames],
|
| 60 |
+
fps=args.fps,
|
| 61 |
+
output_path=args.output,
|
| 62 |
+
audio=audio[0].float().cpu(),
|
| 63 |
+
audio_sample_rate=pipe.vocoder.config.output_sampling_rate,
|
| 64 |
+
)
|
| 65 |
+
print(f"Saved to: {args.output}")
|
model_index.json
ADDED
|
@@ -0,0 +1,36 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"_class_name": "LTX2Pipeline",
|
| 3 |
+
"_diffusers_version": "0.37.0.dev0",
|
| 4 |
+
"audio_vae": [
|
| 5 |
+
"diffusers",
|
| 6 |
+
"AutoencoderKLLTX2Audio"
|
| 7 |
+
],
|
| 8 |
+
"connectors": [
|
| 9 |
+
"ltx2",
|
| 10 |
+
"LTX2TextConnectors"
|
| 11 |
+
],
|
| 12 |
+
"scheduler": [
|
| 13 |
+
"diffusers",
|
| 14 |
+
"FlowMatchEulerDiscreteScheduler"
|
| 15 |
+
],
|
| 16 |
+
"text_encoder": [
|
| 17 |
+
"transformers",
|
| 18 |
+
"Gemma3ForConditionalGeneration"
|
| 19 |
+
],
|
| 20 |
+
"tokenizer": [
|
| 21 |
+
"transformers",
|
| 22 |
+
"GemmaTokenizerFast"
|
| 23 |
+
],
|
| 24 |
+
"transformer": [
|
| 25 |
+
"diffusers",
|
| 26 |
+
"LTX2VideoTransformer3DModel"
|
| 27 |
+
],
|
| 28 |
+
"vae": [
|
| 29 |
+
"diffusers",
|
| 30 |
+
"AutoencoderKLLTX2Video"
|
| 31 |
+
],
|
| 32 |
+
"vocoder": [
|
| 33 |
+
"ltx2",
|
| 34 |
+
"LTX2VocoderWithBWE"
|
| 35 |
+
]
|
| 36 |
+
}
|
processor/added_tokens.json
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"<image_soft_token>": 262144
|
| 3 |
+
}
|
processor/chat_template.jinja
ADDED
|
@@ -0,0 +1,47 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{{ bos_token }}
|
| 2 |
+
{%- if messages[0]['role'] == 'system' -%}
|
| 3 |
+
{%- if messages[0]['content'] is string -%}
|
| 4 |
+
{%- set first_user_prefix = messages[0]['content'] + '
|
| 5 |
+
|
| 6 |
+
' -%}
|
| 7 |
+
{%- else -%}
|
| 8 |
+
{%- set first_user_prefix = messages[0]['content'][0]['text'] + '
|
| 9 |
+
|
| 10 |
+
' -%}
|
| 11 |
+
{%- endif -%}
|
| 12 |
+
{%- set loop_messages = messages[1:] -%}
|
| 13 |
+
{%- else -%}
|
| 14 |
+
{%- set first_user_prefix = "" -%}
|
| 15 |
+
{%- set loop_messages = messages -%}
|
| 16 |
+
{%- endif -%}
|
| 17 |
+
{%- for message in loop_messages -%}
|
| 18 |
+
{%- if (message['role'] == 'user') != (loop.index0 % 2 == 0) -%}
|
| 19 |
+
{{ raise_exception("Conversation roles must alternate user/assistant/user/assistant/...") }}
|
| 20 |
+
{%- endif -%}
|
| 21 |
+
{%- if (message['role'] == 'assistant') -%}
|
| 22 |
+
{%- set role = "model" -%}
|
| 23 |
+
{%- else -%}
|
| 24 |
+
{%- set role = message['role'] -%}
|
| 25 |
+
{%- endif -%}
|
| 26 |
+
{{ '<start_of_turn>' + role + '
|
| 27 |
+
' + (first_user_prefix if loop.first else "") }}
|
| 28 |
+
{%- if message['content'] is string -%}
|
| 29 |
+
{{ message['content'] | trim }}
|
| 30 |
+
{%- elif message['content'] is iterable -%}
|
| 31 |
+
{%- for item in message['content'] -%}
|
| 32 |
+
{%- if item['type'] == 'image' -%}
|
| 33 |
+
{{ '<start_of_image>' }}
|
| 34 |
+
{%- elif item['type'] == 'text' -%}
|
| 35 |
+
{{ item['text'] | trim }}
|
| 36 |
+
{%- endif -%}
|
| 37 |
+
{%- endfor -%}
|
| 38 |
+
{%- else -%}
|
| 39 |
+
{{ raise_exception("Invalid content type") }}
|
| 40 |
+
{%- endif -%}
|
| 41 |
+
{{ '<end_of_turn>
|
| 42 |
+
' }}
|
| 43 |
+
{%- endfor -%}
|
| 44 |
+
{%- if add_generation_prompt -%}
|
| 45 |
+
{{'<start_of_turn>model
|
| 46 |
+
'}}
|
| 47 |
+
{%- endif -%}
|
processor/preprocessor_config.json
ADDED
|
@@ -0,0 +1,29 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"do_convert_rgb": null,
|
| 3 |
+
"do_normalize": true,
|
| 4 |
+
"do_pan_and_scan": null,
|
| 5 |
+
"do_rescale": true,
|
| 6 |
+
"do_resize": true,
|
| 7 |
+
"image_mean": [
|
| 8 |
+
0.5,
|
| 9 |
+
0.5,
|
| 10 |
+
0.5
|
| 11 |
+
],
|
| 12 |
+
"image_processor_type": "Gemma3ImageProcessor",
|
| 13 |
+
"image_seq_length": 256,
|
| 14 |
+
"image_std": [
|
| 15 |
+
0.5,
|
| 16 |
+
0.5,
|
| 17 |
+
0.5
|
| 18 |
+
],
|
| 19 |
+
"pan_and_scan_max_num_crops": null,
|
| 20 |
+
"pan_and_scan_min_crop_size": null,
|
| 21 |
+
"pan_and_scan_min_ratio_to_activate": null,
|
| 22 |
+
"processor_class": "Gemma3Processor",
|
| 23 |
+
"resample": 2,
|
| 24 |
+
"rescale_factor": 0.00392156862745098,
|
| 25 |
+
"size": {
|
| 26 |
+
"height": 896,
|
| 27 |
+
"width": 896
|
| 28 |
+
}
|
| 29 |
+
}
|
processor/processor_config.json
ADDED
|
@@ -0,0 +1,4 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"image_seq_length": 256,
|
| 3 |
+
"processor_class": "Gemma3Processor"
|
| 4 |
+
}
|
processor/special_tokens_map.json
ADDED
|
@@ -0,0 +1,33 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"boi_token": "<start_of_image>",
|
| 3 |
+
"bos_token": {
|
| 4 |
+
"content": "<bos>",
|
| 5 |
+
"lstrip": false,
|
| 6 |
+
"normalized": false,
|
| 7 |
+
"rstrip": false,
|
| 8 |
+
"single_word": false
|
| 9 |
+
},
|
| 10 |
+
"eoi_token": "<end_of_image>",
|
| 11 |
+
"eos_token": {
|
| 12 |
+
"content": "<eos>",
|
| 13 |
+
"lstrip": false,
|
| 14 |
+
"normalized": false,
|
| 15 |
+
"rstrip": false,
|
| 16 |
+
"single_word": false
|
| 17 |
+
},
|
| 18 |
+
"image_token": "<image_soft_token>",
|
| 19 |
+
"pad_token": {
|
| 20 |
+
"content": "<pad>",
|
| 21 |
+
"lstrip": false,
|
| 22 |
+
"normalized": false,
|
| 23 |
+
"rstrip": false,
|
| 24 |
+
"single_word": false
|
| 25 |
+
},
|
| 26 |
+
"unk_token": {
|
| 27 |
+
"content": "<unk>",
|
| 28 |
+
"lstrip": false,
|
| 29 |
+
"normalized": false,
|
| 30 |
+
"rstrip": false,
|
| 31 |
+
"single_word": false
|
| 32 |
+
}
|
| 33 |
+
}
|
processor/tokenizer.json
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4667f2089529e8e7657cfb6d1c19910ae71ff5f28aa7ab2ff2763330affad795
|
| 3 |
+
size 33384568
|
processor/tokenizer.model
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1299c11d7cf632ef3b4e11937501358ada021bbdf7c47638d13c0ee982f2e79c
|
| 3 |
+
size 4689074
|
processor/tokenizer_config.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
scheduler/scheduler_config.json
ADDED
|
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"_class_name": "FlowMatchEulerDiscreteScheduler",
|
| 3 |
+
"_diffusers_version": "0.37.0.dev0",
|
| 4 |
+
"base_image_seq_len": 1024,
|
| 5 |
+
"base_shift": 0.95,
|
| 6 |
+
"invert_sigmas": false,
|
| 7 |
+
"max_image_seq_len": 4096,
|
| 8 |
+
"max_shift": 2.05,
|
| 9 |
+
"num_train_timesteps": 1000,
|
| 10 |
+
"shift": 1.0,
|
| 11 |
+
"shift_terminal": 0.1,
|
| 12 |
+
"stochastic_sampling": false,
|
| 13 |
+
"time_shift_type": "exponential",
|
| 14 |
+
"use_beta_sigmas": false,
|
| 15 |
+
"use_dynamic_shifting": true,
|
| 16 |
+
"use_exponential_sigmas": false,
|
| 17 |
+
"use_karras_sigmas": false
|
| 18 |
+
}
|
text_encoder/config.json
ADDED
|
@@ -0,0 +1,114 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"architectures": [
|
| 3 |
+
"Gemma3ForConditionalGeneration"
|
| 4 |
+
],
|
| 5 |
+
"boi_token_index": 255999,
|
| 6 |
+
"dtype": "float32",
|
| 7 |
+
"eoi_token_index": 256000,
|
| 8 |
+
"eos_token_id": [
|
| 9 |
+
1,
|
| 10 |
+
106
|
| 11 |
+
],
|
| 12 |
+
"image_token_index": 262144,
|
| 13 |
+
"initializer_range": 0.02,
|
| 14 |
+
"mm_tokens_per_image": 256,
|
| 15 |
+
"model_type": "gemma3",
|
| 16 |
+
"text_config": {
|
| 17 |
+
"_sliding_window_pattern": 6,
|
| 18 |
+
"attention_bias": false,
|
| 19 |
+
"attention_dropout": 0.0,
|
| 20 |
+
"attn_logit_softcapping": null,
|
| 21 |
+
"cache_implementation": "hybrid",
|
| 22 |
+
"dtype": "float32",
|
| 23 |
+
"final_logit_softcapping": null,
|
| 24 |
+
"head_dim": 256,
|
| 25 |
+
"hidden_activation": "gelu_pytorch_tanh",
|
| 26 |
+
"hidden_size": 3840,
|
| 27 |
+
"initializer_range": 0.02,
|
| 28 |
+
"intermediate_size": 15360,
|
| 29 |
+
"layer_types": [
|
| 30 |
+
"sliding_attention",
|
| 31 |
+
"sliding_attention",
|
| 32 |
+
"sliding_attention",
|
| 33 |
+
"sliding_attention",
|
| 34 |
+
"sliding_attention",
|
| 35 |
+
"full_attention",
|
| 36 |
+
"sliding_attention",
|
| 37 |
+
"sliding_attention",
|
| 38 |
+
"sliding_attention",
|
| 39 |
+
"sliding_attention",
|
| 40 |
+
"sliding_attention",
|
| 41 |
+
"full_attention",
|
| 42 |
+
"sliding_attention",
|
| 43 |
+
"sliding_attention",
|
| 44 |
+
"sliding_attention",
|
| 45 |
+
"sliding_attention",
|
| 46 |
+
"sliding_attention",
|
| 47 |
+
"full_attention",
|
| 48 |
+
"sliding_attention",
|
| 49 |
+
"sliding_attention",
|
| 50 |
+
"sliding_attention",
|
| 51 |
+
"sliding_attention",
|
| 52 |
+
"sliding_attention",
|
| 53 |
+
"full_attention",
|
| 54 |
+
"sliding_attention",
|
| 55 |
+
"sliding_attention",
|
| 56 |
+
"sliding_attention",
|
| 57 |
+
"sliding_attention",
|
| 58 |
+
"sliding_attention",
|
| 59 |
+
"full_attention",
|
| 60 |
+
"sliding_attention",
|
| 61 |
+
"sliding_attention",
|
| 62 |
+
"sliding_attention",
|
| 63 |
+
"sliding_attention",
|
| 64 |
+
"sliding_attention",
|
| 65 |
+
"full_attention",
|
| 66 |
+
"sliding_attention",
|
| 67 |
+
"sliding_attention",
|
| 68 |
+
"sliding_attention",
|
| 69 |
+
"sliding_attention",
|
| 70 |
+
"sliding_attention",
|
| 71 |
+
"full_attention",
|
| 72 |
+
"sliding_attention",
|
| 73 |
+
"sliding_attention",
|
| 74 |
+
"sliding_attention",
|
| 75 |
+
"sliding_attention",
|
| 76 |
+
"sliding_attention",
|
| 77 |
+
"full_attention"
|
| 78 |
+
],
|
| 79 |
+
"max_position_embeddings": 131072,
|
| 80 |
+
"model_type": "gemma3_text",
|
| 81 |
+
"num_attention_heads": 16,
|
| 82 |
+
"num_hidden_layers": 48,
|
| 83 |
+
"num_key_value_heads": 8,
|
| 84 |
+
"query_pre_attn_scalar": 256,
|
| 85 |
+
"rms_norm_eps": 1e-06,
|
| 86 |
+
"rope_local_base_freq": 10000,
|
| 87 |
+
"rope_scaling": {
|
| 88 |
+
"factor": 8.0,
|
| 89 |
+
"rope_type": "linear"
|
| 90 |
+
},
|
| 91 |
+
"rope_theta": 1000000,
|
| 92 |
+
"sliding_window": 1024,
|
| 93 |
+
"sliding_window_pattern": 6,
|
| 94 |
+
"use_bidirectional_attention": false,
|
| 95 |
+
"use_cache": true,
|
| 96 |
+
"vocab_size": 262208
|
| 97 |
+
},
|
| 98 |
+
"transformers_version": "4.57.3",
|
| 99 |
+
"vision_config": {
|
| 100 |
+
"attention_dropout": 0.0,
|
| 101 |
+
"dtype": "float32",
|
| 102 |
+
"hidden_act": "gelu_pytorch_tanh",
|
| 103 |
+
"hidden_size": 1152,
|
| 104 |
+
"image_size": 896,
|
| 105 |
+
"intermediate_size": 4304,
|
| 106 |
+
"layer_norm_eps": 1e-06,
|
| 107 |
+
"model_type": "siglip_vision_model",
|
| 108 |
+
"num_attention_heads": 16,
|
| 109 |
+
"num_channels": 3,
|
| 110 |
+
"num_hidden_layers": 27,
|
| 111 |
+
"patch_size": 14,
|
| 112 |
+
"vision_use_head": false
|
| 113 |
+
}
|
| 114 |
+
}
|
text_encoder/generation_config.json
ADDED
|
@@ -0,0 +1,11 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"cache_implementation": "hybrid",
|
| 3 |
+
"do_sample": true,
|
| 4 |
+
"eos_token_id": [
|
| 5 |
+
1,
|
| 6 |
+
106
|
| 7 |
+
],
|
| 8 |
+
"top_k": 64,
|
| 9 |
+
"top_p": 0.95,
|
| 10 |
+
"transformers_version": "4.57.3"
|
| 11 |
+
}
|
text_encoder/model-00001-of-00011.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:cbc6e8132e4998652d4c8c219f0ca8da10a143606c806c8728a7513d49bbcdb3
|
| 3 |
+
size 1685223128
|
text_encoder/model-00002-of-00011.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:b95e7ab472b88860e53c71ef078fe8ba4c85e3727b8f0a0bcc130c1f4a2b9ab9
|
| 3 |
+
size 4987027384
|
text_encoder/model-00003-of-00011.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:3731e7c18280d0e9dcbcfadaecaeb8b81fa7f7f15e702ea3d1b4f8eb1b5919f7
|
| 3 |
+
size 4844749824
|
text_encoder/model-00004-of-00011.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e9d1ce8b472f2cc6d70c7885388f50fb3a5f233cf1d4784f6a4be1732547a74c
|
| 3 |
+
size 4954909736
|
text_encoder/model-00005-of-00011.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:cb478659a67b2e34a920f237387a1d7a8208325aa2fc285198a97603f15af1e6
|
| 3 |
+
size 4907664584
|
text_encoder/model-00006-of-00011.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a190581d871980f9309f40d5c2b4db99d76737d7ed743d531bac95443f6b7145
|
| 3 |
+
size 4954909792
|
text_encoder/model-00007-of-00011.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c347de789ff34be642e4fe7ab8e142ca5e3d833d70cb9ec5127a86af0e2ecfed
|
| 3 |
+
size 4907664584
|
text_encoder/model-00008-of-00011.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:2ec7525b89b083d774f4abbdd1f2d7ee190f0475658875cf700530a2faa84a4f
|
| 3 |
+
size 4954909792
|
text_encoder/model-00009-of-00011.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:2b0117ecf1d83691bb875c66a9f2b47e450a87fdfe90915bc7b9e9946c37a650
|
| 3 |
+
size 4907664584
|
text_encoder/model-00010-of-00011.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d9f665a743589231adb812c61c9d8e295c9a68c31d3e87976ae032f1de4e46b6
|
| 3 |
+
size 4954909792
|
text_encoder/model-00011-of-00011.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:999bf4706d4f616e558eec486667fc2b66f8d0f9c106e1d1f802458fb349b0db
|
| 3 |
+
size 2689808472
|
text_encoder/model.safetensors.index.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
tokenizer/added_tokens.json
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"<image_soft_token>": 262144
|
| 3 |
+
}
|
tokenizer/chat_template.jinja
ADDED
|
@@ -0,0 +1,47 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{{ bos_token }}
|
| 2 |
+
{%- if messages[0]['role'] == 'system' -%}
|
| 3 |
+
{%- if messages[0]['content'] is string -%}
|
| 4 |
+
{%- set first_user_prefix = messages[0]['content'] + '
|
| 5 |
+
|
| 6 |
+
' -%}
|
| 7 |
+
{%- else -%}
|
| 8 |
+
{%- set first_user_prefix = messages[0]['content'][0]['text'] + '
|
| 9 |
+
|
| 10 |
+
' -%}
|
| 11 |
+
{%- endif -%}
|
| 12 |
+
{%- set loop_messages = messages[1:] -%}
|
| 13 |
+
{%- else -%}
|
| 14 |
+
{%- set first_user_prefix = "" -%}
|
| 15 |
+
{%- set loop_messages = messages -%}
|
| 16 |
+
{%- endif -%}
|
| 17 |
+
{%- for message in loop_messages -%}
|
| 18 |
+
{%- if (message['role'] == 'user') != (loop.index0 % 2 == 0) -%}
|
| 19 |
+
{{ raise_exception("Conversation roles must alternate user/assistant/user/assistant/...") }}
|
| 20 |
+
{%- endif -%}
|
| 21 |
+
{%- if (message['role'] == 'assistant') -%}
|
| 22 |
+
{%- set role = "model" -%}
|
| 23 |
+
{%- else -%}
|
| 24 |
+
{%- set role = message['role'] -%}
|
| 25 |
+
{%- endif -%}
|
| 26 |
+
{{ '<start_of_turn>' + role + '
|
| 27 |
+
' + (first_user_prefix if loop.first else "") }}
|
| 28 |
+
{%- if message['content'] is string -%}
|
| 29 |
+
{{ message['content'] | trim }}
|
| 30 |
+
{%- elif message['content'] is iterable -%}
|
| 31 |
+
{%- for item in message['content'] -%}
|
| 32 |
+
{%- if item['type'] == 'image' -%}
|
| 33 |
+
{{ '<start_of_image>' }}
|
| 34 |
+
{%- elif item['type'] == 'text' -%}
|
| 35 |
+
{{ item['text'] | trim }}
|
| 36 |
+
{%- endif -%}
|
| 37 |
+
{%- endfor -%}
|
| 38 |
+
{%- else -%}
|
| 39 |
+
{{ raise_exception("Invalid content type") }}
|
| 40 |
+
{%- endif -%}
|
| 41 |
+
{{ '<end_of_turn>
|
| 42 |
+
' }}
|
| 43 |
+
{%- endfor -%}
|
| 44 |
+
{%- if add_generation_prompt -%}
|
| 45 |
+
{{'<start_of_turn>model
|
| 46 |
+
'}}
|
| 47 |
+
{%- endif -%}
|
tokenizer/special_tokens_map.json
ADDED
|
@@ -0,0 +1,33 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"boi_token": "<start_of_image>",
|
| 3 |
+
"bos_token": {
|
| 4 |
+
"content": "<bos>",
|
| 5 |
+
"lstrip": false,
|
| 6 |
+
"normalized": false,
|
| 7 |
+
"rstrip": false,
|
| 8 |
+
"single_word": false
|
| 9 |
+
},
|
| 10 |
+
"eoi_token": "<end_of_image>",
|
| 11 |
+
"eos_token": {
|
| 12 |
+
"content": "<eos>",
|
| 13 |
+
"lstrip": false,
|
| 14 |
+
"normalized": false,
|
| 15 |
+
"rstrip": false,
|
| 16 |
+
"single_word": false
|
| 17 |
+
},
|
| 18 |
+
"image_token": "<image_soft_token>",
|
| 19 |
+
"pad_token": {
|
| 20 |
+
"content": "<pad>",
|
| 21 |
+
"lstrip": false,
|
| 22 |
+
"normalized": false,
|
| 23 |
+
"rstrip": false,
|
| 24 |
+
"single_word": false
|
| 25 |
+
},
|
| 26 |
+
"unk_token": {
|
| 27 |
+
"content": "<unk>",
|
| 28 |
+
"lstrip": false,
|
| 29 |
+
"normalized": false,
|
| 30 |
+
"rstrip": false,
|
| 31 |
+
"single_word": false
|
| 32 |
+
}
|
| 33 |
+
}
|
tokenizer/tokenizer.json
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4667f2089529e8e7657cfb6d1c19910ae71ff5f28aa7ab2ff2763330affad795
|
| 3 |
+
size 33384568
|
tokenizer/tokenizer.model
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1299c11d7cf632ef3b4e11937501358ada021bbdf7c47638d13c0ee982f2e79c
|
| 3 |
+
size 4689074
|
tokenizer/tokenizer_config.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
transformer/config.json
ADDED
|
@@ -0,0 +1,50 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"_class_name": "LTX2VideoTransformer3DModel",
|
| 3 |
+
"_diffusers_version": "0.37.0.dev0",
|
| 4 |
+
"activation_fn": "gelu-approximate",
|
| 5 |
+
"attention_bias": true,
|
| 6 |
+
"attention_head_dim": 128,
|
| 7 |
+
"attention_out_bias": true,
|
| 8 |
+
"audio_attention_head_dim": 64,
|
| 9 |
+
"audio_cross_attention_dim": 2048,
|
| 10 |
+
"audio_cross_attn_mod": true,
|
| 11 |
+
"audio_gated_attn": true,
|
| 12 |
+
"audio_hop_length": 160,
|
| 13 |
+
"audio_in_channels": 128,
|
| 14 |
+
"audio_num_attention_heads": 32,
|
| 15 |
+
"audio_out_channels": 128,
|
| 16 |
+
"audio_patch_size": 1,
|
| 17 |
+
"audio_patch_size_t": 1,
|
| 18 |
+
"audio_pos_embed_max_pos": 20,
|
| 19 |
+
"audio_sampling_rate": 16000,
|
| 20 |
+
"audio_scale_factor": 4,
|
| 21 |
+
"base_height": 2048,
|
| 22 |
+
"base_width": 2048,
|
| 23 |
+
"caption_channels": 3840,
|
| 24 |
+
"causal_offset": 1,
|
| 25 |
+
"cross_attention_dim": 4096,
|
| 26 |
+
"cross_attn_mod": true,
|
| 27 |
+
"cross_attn_timestep_scale_multiplier": 1000,
|
| 28 |
+
"gated_attn": true,
|
| 29 |
+
"in_channels": 128,
|
| 30 |
+
"norm_elementwise_affine": false,
|
| 31 |
+
"norm_eps": 1e-06,
|
| 32 |
+
"num_attention_heads": 32,
|
| 33 |
+
"num_layers": 48,
|
| 34 |
+
"out_channels": 128,
|
| 35 |
+
"patch_size": 1,
|
| 36 |
+
"patch_size_t": 1,
|
| 37 |
+
"perturbed_attn": true,
|
| 38 |
+
"pos_embed_max_pos": 20,
|
| 39 |
+
"qk_norm": "rms_norm_across_heads",
|
| 40 |
+
"rope_double_precision": true,
|
| 41 |
+
"rope_theta": 10000.0,
|
| 42 |
+
"rope_type": "split",
|
| 43 |
+
"timestep_scale_multiplier": 1000,
|
| 44 |
+
"use_prompt_embeddings": false,
|
| 45 |
+
"vae_scale_factors": [
|
| 46 |
+
8,
|
| 47 |
+
32,
|
| 48 |
+
32
|
| 49 |
+
]
|
| 50 |
+
}
|
transformer/diffusion_pytorch_model-00001-of-00008.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:56d3fe084a34b09705a26fb7b8afe319305589d1c034c9da1cd8df1e802b4195
|
| 3 |
+
size 4990709464
|
transformer/diffusion_pytorch_model-00002-of-00008.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:475c1864d130da721d65063ef5f9a7b87f592b0fa82a1895bbb3498ae5729b84
|
| 3 |
+
size 4944487248
|
transformer/diffusion_pytorch_model-00003-of-00008.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:0b1490bd335b22a82a39483d529b9b36d14c8e33e93022e791a67cbce2333e48
|
| 3 |
+
size 4994661712
|
transformer/diffusion_pytorch_model-00004-of-00008.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:ef755b082e189e29acc816513a1102d393b6f9d3885bc43826d7f4203be107fc
|
| 3 |
+
size 4927859352
|
transformer/diffusion_pytorch_model-00005-of-00008.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:b44ce733d707dfd594046829ed53400b192c77715cc44e47186054b8cb633e71
|
| 3 |
+
size 4977604280
|
transformer/diffusion_pytorch_model-00006-of-00008.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:27c3608162c0c5c8202552dcf7f3fd81a0fd13f7ec4616ed7184c8308daa1b34
|
| 3 |
+
size 4994844576
|
transformer/diffusion_pytorch_model-00007-of-00008.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a9b843ebeea2a5e80dd81736c63de83da83d7d472fca62200c6773902c58ed41
|
| 3 |
+
size 4994386864
|
transformer/diffusion_pytorch_model-00008-of-00008.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:99e0201d8630c6c0b8fea1752eb9d8ebe7272c01e7b6c206c7b9e78ece42ce2e
|
| 3 |
+
size 3163094784
|
transformer/diffusion_pytorch_model.safetensors.index.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|