AlphaBrainGroup commited on
Commit
d8a5f51
·
verified ·
1 Parent(s): 480ecd7

Upload neurovla-libero-all4suite

Browse files
.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
+ }