How to use from the
Use from the
Diffusers library
pip install -U diffusers transformers accelerate
import torch
from diffusers import DiffusionPipeline

# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("s1ghhh/VLADrop_GigaBrain0_LIBERO_Baseline", dtype=torch.bfloat16, device_map="cuda")

prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
image = pipe(prompt).images[0]

VLADrop-GigaBrain0-LIBERO-baseline

Checkpoint for Drop-Then-Recovery: How Redundant Are Vision-Language-Action Models?.

DTR (Drop-Then-Recovery) removes transformer blocks from a pretrained VLA model and recovery-fine-tunes the smaller dense model. Code: https://github.com/s1ghhh/VLADrop

This checkpoint

Paper row Table 7: GigaBrain-0 baseline (0/26 dropped)
Dropped blocks none (full model)
Recovery training batch size 16, 50K steps, lr 2.5e-5, fine-tuned from GigaBrain-0-3.5B-Base on standard LIBERO
LIBERO success rate Spatial 84.4 / Object 98.2 / Goal 93.0 / Long 76.2 / Avg 88.0

Usage

GigaBrain-0 checkpoint (model/ format of the giga_models package). Use with the VLADrop giga-brain-0 eval code (https://github.com/s1ghhh/VLADrop):

python 0_eval/run_libero_eval.py \
    --model_path <this_repo_local_path> \
    --norm_stats_path norm_stats_gigabrain.json \
    --task_suite_name libero_spatial --num_trials_per_task 50 --replan_steps 5 \
    [--llm_drop_attn_list ... --llm_drop_mlp_list ...]

Important: for dropped variants, pass the exact drop lists shown above at load time (the drop is applied by a runtime patch, not stored in the weights). norm_stats_gigabrain.json (LIBERO norm stats) is included in this repo.

Citation

@article{sun2026vladrop,
  title={Drop-Then-Recovery: How Redundant Are Vision-Language-Action Models?},
  author={Sun, Guoheng and Feng, Kaixi and He, Shwai and Gong, Xiaochuan and He, Yexiao and Wang, Ziyao and Shen, Zheyu and Ye, Wanghao and Kompella, Ramana Rao and Liu, Gaowen and Li, Ang},
  journal={arXiv preprint arXiv:2606.27755},
  year={2026}
}
Downloads last month
20
Video Preview
loading

Collection including s1ghhh/VLADrop_GigaBrain0_LIBERO_Baseline

Paper for s1ghhh/VLADrop_GigaBrain0_LIBERO_Baseline