Upload neurovla-libero-all4suite
Browse files- .gitattributes +1 -0
- README.md +116 -0
- dataset_statistics.json +134 -0
- framework_config.yaml +97 -0
- model.safetensors +3 -0
- qwen_pretrained/added_tokens.json +24 -0
- qwen_pretrained/chat_template.jinja +7 -0
- qwen_pretrained/config.json +139 -0
- qwen_pretrained/merges.txt +0 -0
- qwen_pretrained/preprocessor_config.json +39 -0
- qwen_pretrained/special_tokens_map.json +31 -0
- qwen_pretrained/tokenizer.json +3 -0
- qwen_pretrained/tokenizer_config.json +208 -0
- qwen_pretrained/video_preprocessor_config.json +43 -0
- qwen_pretrained/vocab.json +0 -0
- resume_meta.json +8 -0
.gitattributes
CHANGED
|
@@ -33,3 +33,4 @@ 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 |
+
qwen_pretrained/tokenizer.json filter=lfs diff=lfs merge=lfs -text
|
README.md
ADDED
|
@@ -0,0 +1,116 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: mit
|
| 3 |
+
language:
|
| 4 |
+
- en
|
| 5 |
+
library_name: pytorch
|
| 6 |
+
pipeline_tag: robotics
|
| 7 |
+
tags:
|
| 8 |
+
- robotics
|
| 9 |
+
- vla
|
| 10 |
+
- vision-language-action
|
| 11 |
+
- libero
|
| 12 |
+
- neurovla
|
| 13 |
+
- spiking-neural-network
|
| 14 |
+
- brain-inspired
|
| 15 |
+
base_model:
|
| 16 |
+
- Qwen/Qwen2.5-VL-3B-Instruct
|
| 17 |
+
datasets:
|
| 18 |
+
- LIBERO
|
| 19 |
+
---
|
| 20 |
+
|
| 21 |
+
# NeuroVLA · LIBERO (all 4 suites, joint training)
|
| 22 |
+
|
| 23 |
+
> Brain-inspired Vision-Language-Action (VLA) checkpoint released with
|
| 24 |
+
> the [AlphaBrain](https://github.com/AlphaBrainGroup/AlphaBrain)
|
| 25 |
+
> framework. Trained jointly on **all four LIBERO suites** — Goal,
|
| 26 |
+
> Spatial, Object, and Long — for direct evaluation across the full
|
| 27 |
+
> LIBERO benchmark without retraining.
|
| 28 |
+
|
| 29 |
+
NeuroVLA couples a Qwen2.5-VL backbone with a **layer-wise Q-Former**
|
| 30 |
+
that extracts action-relevant features from the VLM's hidden states,
|
| 31 |
+
feeding a **Spiking Neural Network (SNN)** action head. The model was
|
| 32 |
+
trained in a single supervised run (not continual learning) on a mixed
|
| 33 |
+
stream of all 4 LIBERO suites, using the `libero_all` data mix.
|
| 34 |
+
|
| 35 |
+
## Overview
|
| 36 |
+
|
| 37 |
+
| | |
|
| 38 |
+
|:---|:---|
|
| 39 |
+
| **Architecture** | NeuroVLA (Qwen2.5-VL-3B + layer-wise Q-Former + SNN head) |
|
| 40 |
+
| **Base VLM** | `Qwen/Qwen2.5-VL-3B-Instruct` |
|
| 41 |
+
| **Q-Former** | Layers 36 → 37 · `num_query_tokens=8` · `output_dim=768` |
|
| 42 |
+
| **Action head** | DiT-based, `hidden_size=1024`, `action_dim=7`, `state_dim=7`, chunk 16 |
|
| 43 |
+
| **Training data** | LIBERO · **all 4 suites (Goal + Spatial + Object + Long)** · `dataset_mix=libero_all` |
|
| 44 |
+
| **Training type** | Supervised fine-tuning (single run; not continual learning) |
|
| 45 |
+
| **Attention** | SDPA (not flash-attention, to avoid ABI pinning) |
|
| 46 |
+
| **Optimiser** | AdamW · `lr_base = 2.5e-5` · cosine-with-min-lr · 5 000 warmup |
|
| 47 |
+
| **Step budget** | 50 000 (this release) · saved every 10 000 steps |
|
| 48 |
+
| **Hardware / batch** | 2 × A800 80 GB · `per_device_batch_size = 16` |
|
| 49 |
+
|
| 50 |
+
## Files
|
| 51 |
+
|
| 52 |
+
```
|
| 53 |
+
├── README.md model card
|
| 54 |
+
├── framework_config.yaml AlphaBrain framework configuration
|
| 55 |
+
├── dataset_statistics.json action normalisation statistics (required for inference)
|
| 56 |
+
├── model.safetensors full VLA weights (~7.7 GB)
|
| 57 |
+
├── resume_meta.json training metadata (step count, GPU count)
|
| 58 |
+
└── qwen_pretrained/ Qwen2.5-VL tokenizer + preprocessor configs
|
| 59 |
+
```
|
| 60 |
+
|
| 61 |
+
## Usage
|
| 62 |
+
|
| 63 |
+
```bash
|
| 64 |
+
git clone https://github.com/AlphaBrainGroup/AlphaBrain.git
|
| 65 |
+
cd AlphaBrain
|
| 66 |
+
pip install -e .
|
| 67 |
+
|
| 68 |
+
export PRETRAINED_MODELS_DIR=/path/to/models # must contain Qwen2.5-VL-3B-Instruct/
|
| 69 |
+
|
| 70 |
+
huggingface-cli download AlphaBrainGroup/neurovla-libero-all4suite \
|
| 71 |
+
--local-dir ./neurovla_libero_all
|
| 72 |
+
|
| 73 |
+
python deployment/model_server/server_policy.py \
|
| 74 |
+
--ckpt_path ./neurovla_libero_all --port 10093 --use_bf16
|
| 75 |
+
```
|
| 76 |
+
|
| 77 |
+
For evaluation on any of the 4 LIBERO suites, see the
|
| 78 |
+
[LIBERO eval pipeline](https://github.com/AlphaBrainGroup/AlphaBrain/tree/dev/benchmarks/LIBERO/eval).
|
| 79 |
+
|
| 80 |
+
## Reproduction
|
| 81 |
+
|
| 82 |
+
```bash
|
| 83 |
+
# Framework's NeuroVLA pretraining entry
|
| 84 |
+
bash scripts/run_brain_inspired_scripts/run_neurovla_pretrain.sh \
|
| 85 |
+
--yaml configs/neurovla_all4suite_libero.yaml # (or equivalent config for 4-suite mix)
|
| 86 |
+
```
|
| 87 |
+
|
| 88 |
+
Expect multi-day training on 2 × A800 80 GB for the full 50 000-step
|
| 89 |
+
schedule. The shipped `framework_config.yaml` is the exact training
|
| 90 |
+
configuration used for this checkpoint.
|
| 91 |
+
|
| 92 |
+
## Notes
|
| 93 |
+
|
| 94 |
+
- **Joint-training baseline**, not continual learning. For the CL
|
| 95 |
+
release of NeuroVLA (sequential training on LIBERO-Goal with
|
| 96 |
+
Experience Replay), see
|
| 97 |
+
[`AlphaBrainGroup/neurovla-cl-libero-goal`](https://huggingface.co/AlphaBrainGroup/neurovla-cl-libero-goal)
|
| 98 |
+
and its LoRA variant.
|
| 99 |
+
- **Attention implementation is SDPA**, chosen to avoid flash-attn ABI
|
| 100 |
+
pinning across environments. Users who have a matching flash-attn
|
| 101 |
+
wheel can override via `--framework.qwenvl.attn_implementation=flash_attention_2`.
|
| 102 |
+
|
| 103 |
+
## License
|
| 104 |
+
|
| 105 |
+
MIT — see the [parent repository](https://github.com/AlphaBrainGroup/AlphaBrain).
|
| 106 |
+
|
| 107 |
+
## Citation
|
| 108 |
+
|
| 109 |
+
```bibtex
|
| 110 |
+
@misc{alphabrain2026,
|
| 111 |
+
title = {AlphaBrain: A Modular Open-Source Framework for Embodied Intelligence Research},
|
| 112 |
+
author = {AlphaBrain Team},
|
| 113 |
+
year = {2026},
|
| 114 |
+
url = {https://github.com/AlphaBrainGroup/AlphaBrain}
|
| 115 |
+
}
|
| 116 |
+
```
|
dataset_statistics.json
ADDED
|
@@ -0,0 +1,134 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"franka": {
|
| 3 |
+
"action": {
|
| 4 |
+
"mean": [
|
| 5 |
+
0.07237596483901143,
|
| 6 |
+
0.08987006871029735,
|
| 7 |
+
-0.10144743137061596,
|
| 8 |
+
-0.00045383188989944756,
|
| 9 |
+
0.006273590726777911,
|
| 10 |
+
-0.003878799732774496,
|
| 11 |
+
0.524486355483532
|
| 12 |
+
],
|
| 13 |
+
"std": [
|
| 14 |
+
0.3498823308902479,
|
| 15 |
+
0.37794140366375184,
|
| 16 |
+
0.460084266976933,
|
| 17 |
+
0.0403885784928603,
|
| 18 |
+
0.06616144248501059,
|
| 19 |
+
0.07763074391911857,
|
| 20 |
+
0.4994683356809767
|
| 21 |
+
],
|
| 22 |
+
"max": [
|
| 23 |
+
0.9375,
|
| 24 |
+
0.9375,
|
| 25 |
+
0.9375,
|
| 26 |
+
0.3557142913341522,
|
| 27 |
+
0.375,
|
| 28 |
+
0.375,
|
| 29 |
+
1.0
|
| 30 |
+
],
|
| 31 |
+
"min": [
|
| 32 |
+
-0.9375,
|
| 33 |
+
-0.9375,
|
| 34 |
+
-0.9375,
|
| 35 |
+
-0.2582142949104309,
|
| 36 |
+
-0.375,
|
| 37 |
+
-0.3675000071525574,
|
| 38 |
+
0.0
|
| 39 |
+
],
|
| 40 |
+
"q01": [
|
| 41 |
+
-0.8785714507102966,
|
| 42 |
+
-0.8758928775787354,
|
| 43 |
+
-0.9375,
|
| 44 |
+
-0.1510714292526245,
|
| 45 |
+
-0.20678570866584778,
|
| 46 |
+
-0.2742857038974762,
|
| 47 |
+
0.0
|
| 48 |
+
],
|
| 49 |
+
"q99": [
|
| 50 |
+
0.9375,
|
| 51 |
+
0.9107142686843872,
|
| 52 |
+
0.9375,
|
| 53 |
+
0.20357142388820648,
|
| 54 |
+
0.26357144117355347,
|
| 55 |
+
0.375,
|
| 56 |
+
1.0
|
| 57 |
+
],
|
| 58 |
+
"mask": [
|
| 59 |
+
true,
|
| 60 |
+
true,
|
| 61 |
+
true,
|
| 62 |
+
true,
|
| 63 |
+
true,
|
| 64 |
+
true,
|
| 65 |
+
false
|
| 66 |
+
],
|
| 67 |
+
"norm_mode": "q99"
|
| 68 |
+
},
|
| 69 |
+
"state": {
|
| 70 |
+
"mean": [
|
| 71 |
+
-0.04889854742214084,
|
| 72 |
+
0.03689368185587227,
|
| 73 |
+
0.7890402488410473,
|
| 74 |
+
2.9771945476531982,
|
| 75 |
+
-0.1417286954820156,
|
| 76 |
+
-0.11769362539052963,
|
| 77 |
+
0.026436020154505968,
|
| 78 |
+
-0.02665513101965189
|
| 79 |
+
],
|
| 80 |
+
"std": [
|
| 81 |
+
0.10639013941746686,
|
| 82 |
+
0.15115733130675715,
|
| 83 |
+
0.38406895599530033,
|
| 84 |
+
0.3530238395244304,
|
| 85 |
+
0.8227341427331599,
|
| 86 |
+
0.32357567121520087,
|
| 87 |
+
0.014583991652936385,
|
| 88 |
+
0.014467005007200339
|
| 89 |
+
],
|
| 90 |
+
"max": [
|
| 91 |
+
0.21031762659549713,
|
| 92 |
+
0.39128610491752625,
|
| 93 |
+
1.3660105466842651,
|
| 94 |
+
3.6714255809783936,
|
| 95 |
+
3.560650587081909,
|
| 96 |
+
1.386339545249939,
|
| 97 |
+
0.04233968257904053,
|
| 98 |
+
0.0013633022317662835
|
| 99 |
+
],
|
| 100 |
+
"min": [
|
| 101 |
+
-0.4828203022480011,
|
| 102 |
+
-0.3255046010017395,
|
| 103 |
+
0.008128180168569088,
|
| 104 |
+
0.35277295112609863,
|
| 105 |
+
-3.641430377960205,
|
| 106 |
+
-1.842738389968872,
|
| 107 |
+
-0.0013586411951109767,
|
| 108 |
+
-0.042040832340717316
|
| 109 |
+
],
|
| 110 |
+
"q01": [
|
| 111 |
+
-0.42401049643754957,
|
| 112 |
+
-0.2838300323486328,
|
| 113 |
+
0.009925739830359817,
|
| 114 |
+
1.3085840785503386,
|
| 115 |
+
-2.886677579879761,
|
| 116 |
+
-1.1599004411697387,
|
| 117 |
+
0.001503719249740243,
|
| 118 |
+
-0.040336399003863335
|
| 119 |
+
],
|
| 120 |
+
"q99": [
|
| 121 |
+
0.1530261474847791,
|
| 122 |
+
0.3629165390133857,
|
| 123 |
+
1.2910678112506866,
|
| 124 |
+
3.303542451858519,
|
| 125 |
+
2.7496529006957933,
|
| 126 |
+
0.6893712210655194,
|
| 127 |
+
0.040610933862626555,
|
| 128 |
+
-0.0015016929572448147
|
| 129 |
+
]
|
| 130 |
+
},
|
| 131 |
+
"num_transitions": 273465,
|
| 132 |
+
"num_trajectories": 1693
|
| 133 |
+
}
|
| 134 |
+
}
|
framework_config.yaml
ADDED
|
@@ -0,0 +1,97 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
framework:
|
| 2 |
+
name: NeuroVLA
|
| 3 |
+
qwenvl:
|
| 4 |
+
attn_implementation: sdpa
|
| 5 |
+
vl_hidden_dim: 2048
|
| 6 |
+
base_vlm: /share/lipengteng/VLA-Engine-Developer/data/pretrained_models/Qwen2.5-VL-3B-Instruct
|
| 7 |
+
layer_qformer:
|
| 8 |
+
qformer_end_layer: 37
|
| 9 |
+
qformer_start_layer: 36
|
| 10 |
+
num_query_tokens: 8
|
| 11 |
+
input_dim: 2048
|
| 12 |
+
ouptput_dim: 768
|
| 13 |
+
grad_scale: 0.5
|
| 14 |
+
action_model:
|
| 15 |
+
hidden_size: 1024
|
| 16 |
+
add_pos_embed: true
|
| 17 |
+
max_seq_len: 1024
|
| 18 |
+
action_dim: 7
|
| 19 |
+
state_dim: 7
|
| 20 |
+
future_action_window_size: 15
|
| 21 |
+
action_horizon: 16
|
| 22 |
+
past_action_window_size: 0
|
| 23 |
+
repeated_diffusion_steps: 8
|
| 24 |
+
reduce_in_full_precision: true
|
| 25 |
+
trainer:
|
| 26 |
+
enable_gradient_checkpointing: true
|
| 27 |
+
enable_mixed_precision_training: true
|
| 28 |
+
epochs: 100
|
| 29 |
+
eval_interval: 50001
|
| 30 |
+
freeze_modules: ''
|
| 31 |
+
gradient_accumulation_steps: 1
|
| 32 |
+
gradient_clipping: 1.0
|
| 33 |
+
is_resume: false
|
| 34 |
+
learning_rate:
|
| 35 |
+
action_model: 0.0001
|
| 36 |
+
base: 2.5e-05
|
| 37 |
+
qwen_vl_interface: 1.0e-05
|
| 38 |
+
layer_qformer: 5.0e-05
|
| 39 |
+
logging_frequency: 10
|
| 40 |
+
loss_scale:
|
| 41 |
+
vla: 1.0
|
| 42 |
+
vlm: 0.1
|
| 43 |
+
lr_scheduler_type: cosine_with_min_lr
|
| 44 |
+
max_grad_norm: 1.0
|
| 45 |
+
max_train_steps: 50000
|
| 46 |
+
num_warmup_steps: 5000
|
| 47 |
+
optimizer:
|
| 48 |
+
betas:
|
| 49 |
+
- 0.9
|
| 50 |
+
- 0.95
|
| 51 |
+
eps: 1.0e-08
|
| 52 |
+
name: AdamW
|
| 53 |
+
weight_decay: 1.0e-08
|
| 54 |
+
resume_epoch: null
|
| 55 |
+
resume_step: null
|
| 56 |
+
save_interval: 10000
|
| 57 |
+
scheduler_specific_kwargs:
|
| 58 |
+
min_lr: 1.0e-06
|
| 59 |
+
warmup_ratio: 0.1
|
| 60 |
+
weight_decay: 0.0
|
| 61 |
+
environment:
|
| 62 |
+
wandb_mode: online
|
| 63 |
+
wandb_project: vla-engine-benchmark
|
| 64 |
+
wandb_entity: ''
|
| 65 |
+
wandb_base_url: https://api.bandw.top
|
| 66 |
+
num_gpus: 2
|
| 67 |
+
main_process_port: 29500
|
| 68 |
+
nccl:
|
| 69 |
+
ib_hca: mlx5_2,mlx5_3
|
| 70 |
+
blocking_wait: 1
|
| 71 |
+
async_error_handling: 1
|
| 72 |
+
timeout: 10000
|
| 73 |
+
socket_timeout_ms: 360000
|
| 74 |
+
seed: 42
|
| 75 |
+
run_id: 0421-NeuroVLA-All4Suite-bs16-sdpa
|
| 76 |
+
output_root_dir: ./results/training
|
| 77 |
+
datasets:
|
| 78 |
+
vla_data:
|
| 79 |
+
data_root_dir: /share/weiyu/IPEC-COMMUNITY
|
| 80 |
+
dataset_mix: libero_all
|
| 81 |
+
per_device_batch_size: 16
|
| 82 |
+
dataloader_module: lerobot_datasets
|
| 83 |
+
action_type: delta_ee
|
| 84 |
+
sequential_step_sampling: false
|
| 85 |
+
CoT_prompt: Your task is {instruction}. To identify the key objects for your task.
|
| 86 |
+
Locate their bounding boxes in [x1,y1,x2,y2] format.
|
| 87 |
+
CoT_answer: bbox
|
| 88 |
+
default_image_resolution:
|
| 89 |
+
- 3
|
| 90 |
+
- 224
|
| 91 |
+
- 224
|
| 92 |
+
load_all_data_for_training: true
|
| 93 |
+
obs:
|
| 94 |
+
- image_0
|
| 95 |
+
video_backend: torchvision_av
|
| 96 |
+
include_state: true
|
| 97 |
+
output_dir: ./results/training/0421-NeuroVLA-All4Suite-bs16-sdpa
|
model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:af72e06391f6e1d95f538d923bf8a14f2542ca2849637beab260f5fae833493a
|
| 3 |
+
size 8167582462
|
qwen_pretrained/added_tokens.json
ADDED
|
@@ -0,0 +1,24 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"</tool_call>": 151658,
|
| 3 |
+
"<tool_call>": 151657,
|
| 4 |
+
"<|box_end|>": 151649,
|
| 5 |
+
"<|box_start|>": 151648,
|
| 6 |
+
"<|endoftext|>": 151643,
|
| 7 |
+
"<|file_sep|>": 151664,
|
| 8 |
+
"<|fim_middle|>": 151660,
|
| 9 |
+
"<|fim_pad|>": 151662,
|
| 10 |
+
"<|fim_prefix|>": 151659,
|
| 11 |
+
"<|fim_suffix|>": 151661,
|
| 12 |
+
"<|im_end|>": 151645,
|
| 13 |
+
"<|im_start|>": 151644,
|
| 14 |
+
"<|image_pad|>": 151655,
|
| 15 |
+
"<|object_ref_end|>": 151647,
|
| 16 |
+
"<|object_ref_start|>": 151646,
|
| 17 |
+
"<|quad_end|>": 151651,
|
| 18 |
+
"<|quad_start|>": 151650,
|
| 19 |
+
"<|repo_name|>": 151663,
|
| 20 |
+
"<|video_pad|>": 151656,
|
| 21 |
+
"<|vision_end|>": 151653,
|
| 22 |
+
"<|vision_pad|>": 151654,
|
| 23 |
+
"<|vision_start|>": 151652
|
| 24 |
+
}
|
qwen_pretrained/chat_template.jinja
ADDED
|
@@ -0,0 +1,7 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{% set image_count = namespace(value=0) %}{% set video_count = namespace(value=0) %}{% for message in messages %}{% if loop.first and message['role'] != 'system' %}<|im_start|>system
|
| 2 |
+
You are a helpful assistant.<|im_end|>
|
| 3 |
+
{% endif %}<|im_start|>{{ message['role'] }}
|
| 4 |
+
{% if message['content'] is string %}{{ message['content'] }}<|im_end|>
|
| 5 |
+
{% else %}{% for content in message['content'] %}{% if content['type'] == 'image' or 'image' in content or 'image_url' in content %}{% set image_count.value = image_count.value + 1 %}{% if add_vision_id %}Picture {{ image_count.value }}: {% endif %}<|vision_start|><|image_pad|><|vision_end|>{% elif content['type'] == 'video' or 'video' in content %}{% set video_count.value = video_count.value + 1 %}{% if add_vision_id %}Video {{ video_count.value }}: {% endif %}<|vision_start|><|video_pad|><|vision_end|>{% elif 'text' in content %}{{ content['text'] }}{% endif %}{% endfor %}<|im_end|>
|
| 6 |
+
{% endif %}{% endfor %}{% if add_generation_prompt %}<|im_start|>assistant
|
| 7 |
+
{% endif %}
|
qwen_pretrained/config.json
ADDED
|
@@ -0,0 +1,139 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"architectures": [
|
| 3 |
+
"Qwen2_5_VLForConditionalGeneration"
|
| 4 |
+
],
|
| 5 |
+
"attention_dropout": 0.0,
|
| 6 |
+
"bos_token_id": 151643,
|
| 7 |
+
"dtype": "bfloat16",
|
| 8 |
+
"eos_token_id": 151645,
|
| 9 |
+
"hidden_act": "silu",
|
| 10 |
+
"hidden_size": 2048,
|
| 11 |
+
"image_token_id": 151655,
|
| 12 |
+
"initializer_range": 0.02,
|
| 13 |
+
"intermediate_size": 11008,
|
| 14 |
+
"max_position_embeddings": 128000,
|
| 15 |
+
"max_window_layers": 70,
|
| 16 |
+
"model_type": "qwen2_5_vl",
|
| 17 |
+
"num_attention_heads": 16,
|
| 18 |
+
"num_hidden_layers": 36,
|
| 19 |
+
"num_key_value_heads": 2,
|
| 20 |
+
"rms_norm_eps": 1e-06,
|
| 21 |
+
"rope_scaling": {
|
| 22 |
+
"mrope_section": [
|
| 23 |
+
16,
|
| 24 |
+
24,
|
| 25 |
+
24
|
| 26 |
+
],
|
| 27 |
+
"rope_type": "default",
|
| 28 |
+
"type": "default"
|
| 29 |
+
},
|
| 30 |
+
"rope_theta": 1000000.0,
|
| 31 |
+
"sliding_window": 32768,
|
| 32 |
+
"text_config": {
|
| 33 |
+
"_name_or_path": "/share/lipengteng/VLA-Engine-Developer/data/pretrained_models/Qwen2.5-VL-3B-Instruct",
|
| 34 |
+
"architectures": [
|
| 35 |
+
"Qwen2_5_VLForConditionalGeneration"
|
| 36 |
+
],
|
| 37 |
+
"attention_dropout": 0.0,
|
| 38 |
+
"bos_token_id": 151643,
|
| 39 |
+
"dtype": "bfloat16",
|
| 40 |
+
"eos_token_id": 151645,
|
| 41 |
+
"hidden_act": "silu",
|
| 42 |
+
"hidden_size": 2048,
|
| 43 |
+
"initializer_range": 0.02,
|
| 44 |
+
"intermediate_size": 11008,
|
| 45 |
+
"layer_types": [
|
| 46 |
+
"full_attention",
|
| 47 |
+
"full_attention",
|
| 48 |
+
"full_attention",
|
| 49 |
+
"full_attention",
|
| 50 |
+
"full_attention",
|
| 51 |
+
"full_attention",
|
| 52 |
+
"full_attention",
|
| 53 |
+
"full_attention",
|
| 54 |
+
"full_attention",
|
| 55 |
+
"full_attention",
|
| 56 |
+
"full_attention",
|
| 57 |
+
"full_attention",
|
| 58 |
+
"full_attention",
|
| 59 |
+
"full_attention",
|
| 60 |
+
"full_attention",
|
| 61 |
+
"full_attention",
|
| 62 |
+
"full_attention",
|
| 63 |
+
"full_attention",
|
| 64 |
+
"full_attention",
|
| 65 |
+
"full_attention",
|
| 66 |
+
"full_attention",
|
| 67 |
+
"full_attention",
|
| 68 |
+
"full_attention",
|
| 69 |
+
"full_attention",
|
| 70 |
+
"full_attention",
|
| 71 |
+
"full_attention",
|
| 72 |
+
"full_attention",
|
| 73 |
+
"full_attention",
|
| 74 |
+
"full_attention",
|
| 75 |
+
"full_attention",
|
| 76 |
+
"full_attention",
|
| 77 |
+
"full_attention",
|
| 78 |
+
"full_attention",
|
| 79 |
+
"full_attention",
|
| 80 |
+
"full_attention",
|
| 81 |
+
"full_attention"
|
| 82 |
+
],
|
| 83 |
+
"max_position_embeddings": 128000,
|
| 84 |
+
"max_window_layers": 70,
|
| 85 |
+
"model_type": "qwen2_5_vl_text",
|
| 86 |
+
"num_attention_heads": 16,
|
| 87 |
+
"num_hidden_layers": 36,
|
| 88 |
+
"num_key_value_heads": 2,
|
| 89 |
+
"rms_norm_eps": 1e-06,
|
| 90 |
+
"rope_scaling": {
|
| 91 |
+
"mrope_section": [
|
| 92 |
+
16,
|
| 93 |
+
24,
|
| 94 |
+
24
|
| 95 |
+
],
|
| 96 |
+
"rope_type": "default",
|
| 97 |
+
"type": "default"
|
| 98 |
+
},
|
| 99 |
+
"rope_theta": 1000000.0,
|
| 100 |
+
"sliding_window": null,
|
| 101 |
+
"tie_word_embeddings": true,
|
| 102 |
+
"use_cache": true,
|
| 103 |
+
"use_sliding_window": false,
|
| 104 |
+
"vision_token_id": 151654,
|
| 105 |
+
"vocab_size": 151936
|
| 106 |
+
},
|
| 107 |
+
"transformers_version": "4.57.0",
|
| 108 |
+
"use_cache": true,
|
| 109 |
+
"use_sliding_window": false,
|
| 110 |
+
"video_token_id": 151656,
|
| 111 |
+
"vision_config": {
|
| 112 |
+
"depth": 32,
|
| 113 |
+
"fullatt_block_indexes": [
|
| 114 |
+
7,
|
| 115 |
+
15,
|
| 116 |
+
23,
|
| 117 |
+
31
|
| 118 |
+
],
|
| 119 |
+
"hidden_act": "silu",
|
| 120 |
+
"hidden_size": 1280,
|
| 121 |
+
"in_channels": 3,
|
| 122 |
+
"in_chans": 3,
|
| 123 |
+
"initializer_range": 0.02,
|
| 124 |
+
"intermediate_size": 3420,
|
| 125 |
+
"model_type": "qwen2_5_vl",
|
| 126 |
+
"num_heads": 16,
|
| 127 |
+
"out_hidden_size": 2048,
|
| 128 |
+
"patch_size": 14,
|
| 129 |
+
"spatial_merge_size": 2,
|
| 130 |
+
"spatial_patch_size": 14,
|
| 131 |
+
"temporal_patch_size": 2,
|
| 132 |
+
"tokens_per_second": 2,
|
| 133 |
+
"window_size": 112
|
| 134 |
+
},
|
| 135 |
+
"vision_end_token_id": 151653,
|
| 136 |
+
"vision_start_token_id": 151652,
|
| 137 |
+
"vision_token_id": 151654,
|
| 138 |
+
"vocab_size": 151936
|
| 139 |
+
}
|
qwen_pretrained/merges.txt
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
qwen_pretrained/preprocessor_config.json
ADDED
|
@@ -0,0 +1,39 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"crop_size": null,
|
| 3 |
+
"data_format": "channels_first",
|
| 4 |
+
"default_to_square": true,
|
| 5 |
+
"device": null,
|
| 6 |
+
"disable_grouping": null,
|
| 7 |
+
"do_center_crop": null,
|
| 8 |
+
"do_convert_rgb": true,
|
| 9 |
+
"do_normalize": true,
|
| 10 |
+
"do_pad": null,
|
| 11 |
+
"do_rescale": true,
|
| 12 |
+
"do_resize": true,
|
| 13 |
+
"image_mean": [
|
| 14 |
+
0.48145466,
|
| 15 |
+
0.4578275,
|
| 16 |
+
0.40821073
|
| 17 |
+
],
|
| 18 |
+
"image_processor_type": "Qwen2VLImageProcessorFast",
|
| 19 |
+
"image_std": [
|
| 20 |
+
0.26862954,
|
| 21 |
+
0.26130258,
|
| 22 |
+
0.27577711
|
| 23 |
+
],
|
| 24 |
+
"input_data_format": null,
|
| 25 |
+
"max_pixels": 12845056,
|
| 26 |
+
"merge_size": 2,
|
| 27 |
+
"min_pixels": 3136,
|
| 28 |
+
"pad_size": null,
|
| 29 |
+
"patch_size": 14,
|
| 30 |
+
"processor_class": "Qwen2_5_VLProcessor",
|
| 31 |
+
"resample": 3,
|
| 32 |
+
"rescale_factor": 0.00392156862745098,
|
| 33 |
+
"return_tensors": null,
|
| 34 |
+
"size": {
|
| 35 |
+
"longest_edge": 12845056,
|
| 36 |
+
"shortest_edge": 3136
|
| 37 |
+
},
|
| 38 |
+
"temporal_patch_size": 2
|
| 39 |
+
}
|
qwen_pretrained/special_tokens_map.json
ADDED
|
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"additional_special_tokens": [
|
| 3 |
+
"<|im_start|>",
|
| 4 |
+
"<|im_end|>",
|
| 5 |
+
"<|object_ref_start|>",
|
| 6 |
+
"<|object_ref_end|>",
|
| 7 |
+
"<|box_start|>",
|
| 8 |
+
"<|box_end|>",
|
| 9 |
+
"<|quad_start|>",
|
| 10 |
+
"<|quad_end|>",
|
| 11 |
+
"<|vision_start|>",
|
| 12 |
+
"<|vision_end|>",
|
| 13 |
+
"<|vision_pad|>",
|
| 14 |
+
"<|image_pad|>",
|
| 15 |
+
"<|video_pad|>"
|
| 16 |
+
],
|
| 17 |
+
"eos_token": {
|
| 18 |
+
"content": "<|im_end|>",
|
| 19 |
+
"lstrip": false,
|
| 20 |
+
"normalized": false,
|
| 21 |
+
"rstrip": false,
|
| 22 |
+
"single_word": false
|
| 23 |
+
},
|
| 24 |
+
"pad_token": {
|
| 25 |
+
"content": "<|endoftext|>",
|
| 26 |
+
"lstrip": false,
|
| 27 |
+
"normalized": false,
|
| 28 |
+
"rstrip": false,
|
| 29 |
+
"single_word": false
|
| 30 |
+
}
|
| 31 |
+
}
|
qwen_pretrained/tokenizer.json
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:5eee858c5123a4279c3e1f7b81247343f356ac767940b2692a928ad929543214
|
| 3 |
+
size 11422063
|
qwen_pretrained/tokenizer_config.json
ADDED
|
@@ -0,0 +1,208 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"add_bos_token": false,
|
| 3 |
+
"add_prefix_space": false,
|
| 4 |
+
"added_tokens_decoder": {
|
| 5 |
+
"151643": {
|
| 6 |
+
"content": "<|endoftext|>",
|
| 7 |
+
"lstrip": false,
|
| 8 |
+
"normalized": false,
|
| 9 |
+
"rstrip": false,
|
| 10 |
+
"single_word": false,
|
| 11 |
+
"special": true
|
| 12 |
+
},
|
| 13 |
+
"151644": {
|
| 14 |
+
"content": "<|im_start|>",
|
| 15 |
+
"lstrip": false,
|
| 16 |
+
"normalized": false,
|
| 17 |
+
"rstrip": false,
|
| 18 |
+
"single_word": false,
|
| 19 |
+
"special": true
|
| 20 |
+
},
|
| 21 |
+
"151645": {
|
| 22 |
+
"content": "<|im_end|>",
|
| 23 |
+
"lstrip": false,
|
| 24 |
+
"normalized": false,
|
| 25 |
+
"rstrip": false,
|
| 26 |
+
"single_word": false,
|
| 27 |
+
"special": true
|
| 28 |
+
},
|
| 29 |
+
"151646": {
|
| 30 |
+
"content": "<|object_ref_start|>",
|
| 31 |
+
"lstrip": false,
|
| 32 |
+
"normalized": false,
|
| 33 |
+
"rstrip": false,
|
| 34 |
+
"single_word": false,
|
| 35 |
+
"special": true
|
| 36 |
+
},
|
| 37 |
+
"151647": {
|
| 38 |
+
"content": "<|object_ref_end|>",
|
| 39 |
+
"lstrip": false,
|
| 40 |
+
"normalized": false,
|
| 41 |
+
"rstrip": false,
|
| 42 |
+
"single_word": false,
|
| 43 |
+
"special": true
|
| 44 |
+
},
|
| 45 |
+
"151648": {
|
| 46 |
+
"content": "<|box_start|>",
|
| 47 |
+
"lstrip": false,
|
| 48 |
+
"normalized": false,
|
| 49 |
+
"rstrip": false,
|
| 50 |
+
"single_word": false,
|
| 51 |
+
"special": true
|
| 52 |
+
},
|
| 53 |
+
"151649": {
|
| 54 |
+
"content": "<|box_end|>",
|
| 55 |
+
"lstrip": false,
|
| 56 |
+
"normalized": false,
|
| 57 |
+
"rstrip": false,
|
| 58 |
+
"single_word": false,
|
| 59 |
+
"special": true
|
| 60 |
+
},
|
| 61 |
+
"151650": {
|
| 62 |
+
"content": "<|quad_start|>",
|
| 63 |
+
"lstrip": false,
|
| 64 |
+
"normalized": false,
|
| 65 |
+
"rstrip": false,
|
| 66 |
+
"single_word": false,
|
| 67 |
+
"special": true
|
| 68 |
+
},
|
| 69 |
+
"151651": {
|
| 70 |
+
"content": "<|quad_end|>",
|
| 71 |
+
"lstrip": false,
|
| 72 |
+
"normalized": false,
|
| 73 |
+
"rstrip": false,
|
| 74 |
+
"single_word": false,
|
| 75 |
+
"special": true
|
| 76 |
+
},
|
| 77 |
+
"151652": {
|
| 78 |
+
"content": "<|vision_start|>",
|
| 79 |
+
"lstrip": false,
|
| 80 |
+
"normalized": false,
|
| 81 |
+
"rstrip": false,
|
| 82 |
+
"single_word": false,
|
| 83 |
+
"special": true
|
| 84 |
+
},
|
| 85 |
+
"151653": {
|
| 86 |
+
"content": "<|vision_end|>",
|
| 87 |
+
"lstrip": false,
|
| 88 |
+
"normalized": false,
|
| 89 |
+
"rstrip": false,
|
| 90 |
+
"single_word": false,
|
| 91 |
+
"special": true
|
| 92 |
+
},
|
| 93 |
+
"151654": {
|
| 94 |
+
"content": "<|vision_pad|>",
|
| 95 |
+
"lstrip": false,
|
| 96 |
+
"normalized": false,
|
| 97 |
+
"rstrip": false,
|
| 98 |
+
"single_word": false,
|
| 99 |
+
"special": true
|
| 100 |
+
},
|
| 101 |
+
"151655": {
|
| 102 |
+
"content": "<|image_pad|>",
|
| 103 |
+
"lstrip": false,
|
| 104 |
+
"normalized": false,
|
| 105 |
+
"rstrip": false,
|
| 106 |
+
"single_word": false,
|
| 107 |
+
"special": true
|
| 108 |
+
},
|
| 109 |
+
"151656": {
|
| 110 |
+
"content": "<|video_pad|>",
|
| 111 |
+
"lstrip": false,
|
| 112 |
+
"normalized": false,
|
| 113 |
+
"rstrip": false,
|
| 114 |
+
"single_word": false,
|
| 115 |
+
"special": true
|
| 116 |
+
},
|
| 117 |
+
"151657": {
|
| 118 |
+
"content": "<tool_call>",
|
| 119 |
+
"lstrip": false,
|
| 120 |
+
"normalized": false,
|
| 121 |
+
"rstrip": false,
|
| 122 |
+
"single_word": false,
|
| 123 |
+
"special": false
|
| 124 |
+
},
|
| 125 |
+
"151658": {
|
| 126 |
+
"content": "</tool_call>",
|
| 127 |
+
"lstrip": false,
|
| 128 |
+
"normalized": false,
|
| 129 |
+
"rstrip": false,
|
| 130 |
+
"single_word": false,
|
| 131 |
+
"special": false
|
| 132 |
+
},
|
| 133 |
+
"151659": {
|
| 134 |
+
"content": "<|fim_prefix|>",
|
| 135 |
+
"lstrip": false,
|
| 136 |
+
"normalized": false,
|
| 137 |
+
"rstrip": false,
|
| 138 |
+
"single_word": false,
|
| 139 |
+
"special": false
|
| 140 |
+
},
|
| 141 |
+
"151660": {
|
| 142 |
+
"content": "<|fim_middle|>",
|
| 143 |
+
"lstrip": false,
|
| 144 |
+
"normalized": false,
|
| 145 |
+
"rstrip": false,
|
| 146 |
+
"single_word": false,
|
| 147 |
+
"special": false
|
| 148 |
+
},
|
| 149 |
+
"151661": {
|
| 150 |
+
"content": "<|fim_suffix|>",
|
| 151 |
+
"lstrip": false,
|
| 152 |
+
"normalized": false,
|
| 153 |
+
"rstrip": false,
|
| 154 |
+
"single_word": false,
|
| 155 |
+
"special": false
|
| 156 |
+
},
|
| 157 |
+
"151662": {
|
| 158 |
+
"content": "<|fim_pad|>",
|
| 159 |
+
"lstrip": false,
|
| 160 |
+
"normalized": false,
|
| 161 |
+
"rstrip": false,
|
| 162 |
+
"single_word": false,
|
| 163 |
+
"special": false
|
| 164 |
+
},
|
| 165 |
+
"151663": {
|
| 166 |
+
"content": "<|repo_name|>",
|
| 167 |
+
"lstrip": false,
|
| 168 |
+
"normalized": false,
|
| 169 |
+
"rstrip": false,
|
| 170 |
+
"single_word": false,
|
| 171 |
+
"special": false
|
| 172 |
+
},
|
| 173 |
+
"151664": {
|
| 174 |
+
"content": "<|file_sep|>",
|
| 175 |
+
"lstrip": false,
|
| 176 |
+
"normalized": false,
|
| 177 |
+
"rstrip": false,
|
| 178 |
+
"single_word": false,
|
| 179 |
+
"special": false
|
| 180 |
+
}
|
| 181 |
+
},
|
| 182 |
+
"additional_special_tokens": [
|
| 183 |
+
"<|im_start|>",
|
| 184 |
+
"<|im_end|>",
|
| 185 |
+
"<|object_ref_start|>",
|
| 186 |
+
"<|object_ref_end|>",
|
| 187 |
+
"<|box_start|>",
|
| 188 |
+
"<|box_end|>",
|
| 189 |
+
"<|quad_start|>",
|
| 190 |
+
"<|quad_end|>",
|
| 191 |
+
"<|vision_start|>",
|
| 192 |
+
"<|vision_end|>",
|
| 193 |
+
"<|vision_pad|>",
|
| 194 |
+
"<|image_pad|>",
|
| 195 |
+
"<|video_pad|>"
|
| 196 |
+
],
|
| 197 |
+
"bos_token": null,
|
| 198 |
+
"clean_up_tokenization_spaces": false,
|
| 199 |
+
"eos_token": "<|im_end|>",
|
| 200 |
+
"errors": "replace",
|
| 201 |
+
"extra_special_tokens": {},
|
| 202 |
+
"model_max_length": 131072,
|
| 203 |
+
"pad_token": "<|endoftext|>",
|
| 204 |
+
"processor_class": "Qwen2_5_VLProcessor",
|
| 205 |
+
"split_special_tokens": false,
|
| 206 |
+
"tokenizer_class": "Qwen2Tokenizer",
|
| 207 |
+
"unk_token": null
|
| 208 |
+
}
|
qwen_pretrained/video_preprocessor_config.json
ADDED
|
@@ -0,0 +1,43 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"crop_size": null,
|
| 3 |
+
"data_format": "channels_first",
|
| 4 |
+
"default_to_square": true,
|
| 5 |
+
"device": null,
|
| 6 |
+
"do_center_crop": null,
|
| 7 |
+
"do_convert_rgb": true,
|
| 8 |
+
"do_normalize": true,
|
| 9 |
+
"do_rescale": true,
|
| 10 |
+
"do_resize": true,
|
| 11 |
+
"do_sample_frames": false,
|
| 12 |
+
"fps": null,
|
| 13 |
+
"image_mean": [
|
| 14 |
+
0.48145466,
|
| 15 |
+
0.4578275,
|
| 16 |
+
0.40821073
|
| 17 |
+
],
|
| 18 |
+
"image_std": [
|
| 19 |
+
0.26862954,
|
| 20 |
+
0.26130258,
|
| 21 |
+
0.27577711
|
| 22 |
+
],
|
| 23 |
+
"input_data_format": null,
|
| 24 |
+
"max_frames": 768,
|
| 25 |
+
"max_pixels": 12845056,
|
| 26 |
+
"merge_size": 2,
|
| 27 |
+
"min_frames": 4,
|
| 28 |
+
"min_pixels": 3136,
|
| 29 |
+
"num_frames": null,
|
| 30 |
+
"pad_size": null,
|
| 31 |
+
"patch_size": 14,
|
| 32 |
+
"processor_class": "Qwen2_5_VLProcessor",
|
| 33 |
+
"resample": 3,
|
| 34 |
+
"rescale_factor": 0.00392156862745098,
|
| 35 |
+
"return_metadata": false,
|
| 36 |
+
"size": {
|
| 37 |
+
"longest_edge": 12845056,
|
| 38 |
+
"shortest_edge": 3136
|
| 39 |
+
},
|
| 40 |
+
"temporal_patch_size": 2,
|
| 41 |
+
"video_metadata": null,
|
| 42 |
+
"video_processor_type": "Qwen2VLVideoProcessor"
|
| 43 |
+
}
|
qwen_pretrained/vocab.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
resume_meta.json
ADDED
|
@@ -0,0 +1,8 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"completed_steps": 50000,
|
| 3 |
+
"num_gpus": 4,
|
| 4 |
+
"gradient_accumulation_steps": 1,
|
| 5 |
+
"per_device_batch_size": 16,
|
| 6 |
+
"effective_batch_size": 64,
|
| 7 |
+
"framework_name": "NeuroVLA"
|
| 8 |
+
}
|