Instructions to use AAyano/oft_setting1_chunksize25_batch32_10k with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use AAyano/oft_setting1_chunksize25_batch32_10k with Transformers:
# Load model directly from transformers import AutoModelForVision2Seq model = AutoModelForVision2Seq.from_pretrained("AAyano/oft_setting1_chunksize25_batch32_10k", trust_remote_code=True, dtype="auto") - Notebooks
- Google Colab
- Kaggle
oft_setting1_chunksize25_batch32_10k
OpenVLA-OFT checkpoint fine-tuned on real-world XArm data, setting 1: pick cube and place into plastic cup. Intermediate checkpoint at 10000 / 30000 training steps (see oft_setting1_chunksize25_batch32 for the final 30k-step model). LoRA weights (rank 32) are already merged into the openvla/openvla-7b base — this repo is a standalone model.
Training
- Recipe: OpenVLA-OFT (L1 regression, FiLM, parallel decoding)
- Steps: 10000 of 30000 (LR 5e-4, decay after 20000), effective batch size 32
- Action chunk: 25, action dim: 7, proprio dim: 6 (BOUNDS_Q99 normalization)
- Inputs: 2 images (3rd-person + wrist) + proprio
Contents
- Merged model shards (
model-*.safetensors) action_head--10000_checkpoint.pt,proprio_projector--10000_checkpoint.pt,vision_backbone--10000_checkpoint.pt— OFT components needed for deploymentdataset_statistics.json— action/proprio normalization statisticsoft_training_config.json— training configuration snapshot
Use with the openvla-oft codebase; set XArm constants (chunk 25, action dim 7, proprio dim 6) in prismatic/vla/constants.py before deployment.
- Downloads last month
- 14
Model tree for AAyano/oft_setting1_chunksize25_batch32_10k
Base model
openvla/openvla-7b