Qwen3.5-0.8B-FFT-LAP-NOMASK

VLA-0 checkpoint: Qwen/Qwen3.5-0.8B fine-tuned with Full Fine-Tuning on LIBERO benchmark tasks.

VLA-0 represents robot actions directly as text tokens — no architectural changes to the base VLM.

Quick Start

1. Download

pip install huggingface_hub
huggingface-cli download denkiwakame/Qwen3.5-0.8B-FFT-LAP-NOMASK --local-dir ./Qwen3.5-0.8B-FFT-LAP-NOMASK

2. Load with transformers

import pickle
import torch
from transformers import Qwen2_5_VLForConditionalGeneration, Qwen2_5_VLProcessor

ckpt_dir = "./Qwen3.5-0.8B-FFT-LAP-NOMASK"

model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
    f"{ckpt_dir}/model_final", torch_dtype=torch.bfloat16, device_map="auto",
)
processor = Qwen2_5_VLProcessor.from_pretrained(f"{ckpt_dir}/model_final")

# Load dataset stats (required for action denormalization)
with open(f"{ckpt_dir}/dataset_stats.pkl", "rb") as f:
    dataset_stats = pickle.load(f)

3. Load with VLA-0 framework

from rv_train.train import get_pretrained_model

model, cfg = get_pretrained_model("./Qwen3.5-0.8B-FFT-LAP-NOMASK", device=0)
model.eval()

dataset_stats.pkl

Action normalization statistics computed from the training dataset. Required at inference time to denormalize model outputs back to the original action space.

import pickle

with open("dataset_stats.pkl", "rb") as f:
    stats = pickle.load(f)
# stats contains mean/std for action dimensions

Intermediate Checkpoints

main holds the recommended/final weights. Earlier training-step snapshots are published as branches named step-<global_step> (e.g., step-17000, step-18000). Load any of them by passing revision=:

# Download a specific revision
huggingface-cli download denkiwakame/Qwen3.5-0.8B-FFT-LAP-NOMASK --revision step-18000 --local-dir ./Qwen3.5-0.8B-FFT-LAP-NOMASK-step-18000

# Or load directly via transformers
Qwen2_5_VLForConditionalGeneration.from_pretrained(
    "denkiwakame/Qwen3.5-0.8B-FFT-LAP-NOMASK",
    revision="step-18000",
    subfolder="model_final",
)

See the repository branches tab for the full list.

Training Details

  • Base Model: Qwen/Qwen3.5-0.8B
  • Method: Full Fine-Tuning
  • Dataset: LIBERO (via RoboVerse)
  • Framework: VLA-0
Training Config
DATALOADER:
  ROBOVERSE:
    cfg_opts: IMAGE.crop_img:0.875:IMAGE.img_size:224:IMAGE.cam_list:('3p1','3p2')
    cfg_path: libs/RoboVerse/roboverse/configs/img_libero_aug.yaml
  batch_size: 16
  num_workers: 8
EXP:
  AMP: true
  DATASET: roboverse
  EXP_ID: lap_qwen3_5_08b_fft_nomask
  LOSS: {}
  LR_SCHED: none
  MODEL: qwen
  OPTIMIZER: adamw
  SEED: 0
EXP_EXTRA:
  no_test: true
  no_track: true
  no_val: true
  save_ckp: 2
  save_last_ckpt: true
  test_eval_freq: 1
  val_eval_freq: 1
LR_SCHED:
  lr_clip: 1.0e-08
  lr_decay_factor: 0.5
  lr_patience: 4
MODEL:
  QWEN:
    action_mask_aug_per: 0.0
    action_type: original
    add_vision_id: false
    attention_dropout: 0.0
    enable_thinking: true
    grad_checkpoint: false
    history: 1
    horizon: 8
    lora_config: default
    lora_rank: 8
    num_bins_actions: 1000
    num_cam: 2
    original_action_dim: 7
    qwen_model_id: Qwen/Qwen3.5-0.8B
    reasoning: true
    rgb_img_size:
    - 224
    - 224
    rgb_input: true
    tiled_rgb_imgs: true
    use_flash_attention_2: true
    use_lora: false
    use_qlora: false
TRAIN:
  clip_grad_norm: 0.0
  l2: 1.0e-10
  lr: 5.0e-06
  num_epochs: 100
  num_iters: 20000
  save_iter_ckp: 10000
WANDB:
  enable: true
  entity: ''
  example_log_interval: 2000
  mode: online
  project: vla0
  resume_id: ''
  run_name: ''
  tags: ''

Files

File Description
model_final/model-*.safetensors Full model weights
model_final/config.json Model configuration
model_final/tokenizer.json Tokenizer
dataset_stats.pkl Action normalization statistics (required for inference)
config.yaml Training configuration

License

CC-BY-NC-4.0 (following the upstream VLA-0 license). Subject to Qwen License for the base model.

Downloads last month
4
Video Preview
loading

Model tree for denkiwakame/Qwen3.5-0.8B-FFT-LAP-NOMASK

Finetuned
(412)
this model