tribev2-balanced-qv

This repo packages the full TRIBEv2 model with a quantized V-JEPA2 video branch.

The original TRIBEv2 checkpoint is still there, and the quantized video weights are already included in quantized_video/, so the repo stays easy to move around and use as a single package. The bundled loader keeps the original V-JEPA2 backbone id visible to TRIBEv2 consumers and swaps in the embedded quantized weights from quantized_video/ at load time. All presets in this family keep a ViT-G-compatible video feature interface so the untouched TRIBEv2 checkpoint remains dimensionally compatible.

This version

Balanced runtime profile using the same ViT-G-compatible quantized video branch, with moderate sampling and batch size.

Good fit for:

• General-purpose sharing
• Default recommendation
• Users who want a good compromise

In short:

• Quality: Middle profile between lite and high_quality.
• Memory: Moderate runtime pressure.
• Ease of use: Balanced compromise without being the lightest.

Runtime profile

• Profile: Balanced profile with moderate video sampling and batch size.
• Video backbone: facebook/vjepa2-vitg-fpc64-256
• Video event frequency: 1.5 Hz
• Video batch size: 4
• Video num_frames: default

What you get

• Full TRIBEv2 repo in one place
• Original best.ckpt
• Audio, text and fusion parts left as-is
• Quantized video branch in quantized_video/
• Loader helper in load_quantized_tribev2.py that injects the embedded quantized video branch

Quick start

pip install "torch>=2.11" transformers huggingface_hub pyyaml torchao tribev2

Use the bundled loader below rather than calling TribeModel.from_pretrained(...) directly on the repo.

from pathlib import Path
from load_quantized_tribev2 import load_quantized_tribev2

repo_dir = Path(".")
model = load_quantized_tribev2(repo_dir, device="auto")
print(type(model).__name__)

load_quantized_tribev2() keeps data.video_feature.image.model_name aligned with facebook/vjepa2-vitg-fpc64-256 for compatibility and injects the embedded quantized weights from quantized_video/.

Based on

• facebook/tribev2
• facebook/vjepa2-vitg-fpc64-256