TieuDaoChanNhan commited on
Commit
a09203e
·
verified ·
1 Parent(s): 5237620

Upload VLA tokenizer with model card

Browse files
Files changed (2) hide show
  1. README.md +13 -6
  2. tokenizer.json +2 -2
README.md CHANGED
@@ -10,18 +10,24 @@ tags:
10
  license: apache-2.0
11
  ---
12
 
13
- # VLA Tokenizer — Adaptive v2 (GPT-NeoX-20b + SNAC)
14
 
15
  Extended GPT-NeoX-20b tokenizer for the **FineVideo-VLA** multimodal dataset.
16
- Adds 3D human pose tokens, video tokens, and SNAC audio tokens on top of the
 
17
  [EleutherAI/gpt-neox-20b](https://huggingface.co/EleutherAI/gpt-neox-20b) base.
18
 
19
- **Vocab size: 156,505** (50,277 base + 93,938 VLA + 12,290 SNAC)
20
 
21
  > **v1 → v2 change:** Added 12,290 SNAC audio tokens (`<snac>`, `</snac>`,
22
  > and 12,288 `<snac_N>` tokens) for the SNAC listen format used in
23
  > [MixtureVitae-Omni](https://huggingface.co/datasets/mixture-vitae/MixtureVitae-Omni)
24
  > and FineVideo-VLA audio tokenization. All existing v1 token IDs are unchanged.
 
 
 
 
 
25
 
26
  ---
27
 
@@ -41,8 +47,9 @@ Adds 3D human pose tokens, video tokens, and SNAC audio tokens on top of the
41
  | **SNAC Level 1 even** | `<snac_132362>` – `<snac_136457>` | 4,096 | 25 Hz fine audio (even frames) |
42
  | **SNAC Level 1 odd** | `<snac_144650>` – `<snac_148745>` | 4,096 | 25 Hz fine audio (odd frames) |
43
  | **SNAC wrappers** | `<snac>`, `</snac>` | 2 | Block delimiters |
 
44
 
45
- **Total new tokens: 106,228** (93,938 VLA + 12,290 SNAC)
46
 
47
  ---
48
 
@@ -84,7 +91,7 @@ SNAC listen format: 3 tokens per base frame (L0 + L1_even + L1_odd),
84
  from transformers import AutoTokenizer
85
 
86
  tok = AutoTokenizer.from_pretrained("EmpathicRobotics/tokenizer-vla-adaptive-v2")
87
- print(len(tok)) # 156505
88
 
89
  # All VLA and SNAC tokens are single atomic tokens
90
  print(tok.encode("<seed2_1137>", add_special_tokens=False)) # [59908]
@@ -115,7 +122,7 @@ tok.save_pretrained("tokenizer-vla-adaptive-v2")
115
  # vocab size: 156,505
116
  ```
117
 
118
- Script: `tools/build_tokenizers.py` in the
119
  [finevideo-vla](https://github.com/TieuDaoChanNhan/finevideo-vla) repo.
120
 
121
  ---
 
10
  license: apache-2.0
11
  ---
12
 
13
+ # VLA Tokenizer — Adaptive v2 (GPT-NeoX-20b + SNAC + caption/speech)
14
 
15
  Extended GPT-NeoX-20b tokenizer for the **FineVideo-VLA** multimodal dataset.
16
+ Adds 3D human pose tokens, video tokens, SNAC audio tokens, and caption/speech
17
+ wrapper tokens on top of the
18
  [EleutherAI/gpt-neox-20b](https://huggingface.co/EleutherAI/gpt-neox-20b) base.
19
 
20
+ **Vocab size: 156,509** (50,277 base + 93,938 VLA + 12,290 SNAC + 4 caption/speech)
21
 
22
  > **v1 → v2 change:** Added 12,290 SNAC audio tokens (`<snac>`, `</snac>`,
23
  > and 12,288 `<snac_N>` tokens) for the SNAC listen format used in
24
  > [MixtureVitae-Omni](https://huggingface.co/datasets/mixture-vitae/MixtureVitae-Omni)
25
  > and FineVideo-VLA audio tokenization. All existing v1 token IDs are unchanged.
26
+ >
27
+ > **Later addition (same v2 repo, in place):** Added 4 wrapper tokens —
28
+ > `<caption>`, `</caption>`, `<speech>`, `</speech>` — for inline visual
29
+ > caption / spoken-dialogue interleaving at modality-transition points in the
30
+ > token sequence. All prior token IDs (including SNAC) are unchanged.
31
 
32
  ---
33
 
 
47
  | **SNAC Level 1 even** | `<snac_132362>` – `<snac_136457>` | 4,096 | 25 Hz fine audio (even frames) |
48
  | **SNAC Level 1 odd** | `<snac_144650>` – `<snac_148745>` | 4,096 | 25 Hz fine audio (odd frames) |
49
  | **SNAC wrappers** | `<snac>`, `</snac>` | 2 | Block delimiters |
50
+ | **Caption/speech wrappers** | `<caption>`, `</caption>`, `<speech>`, `</speech>` | 4 | Inline caption/dialogue interleaving |
51
 
52
+ **Total new tokens: 106,232** (93,938 VLA + 12,290 SNAC + 4 caption/speech)
53
 
54
  ---
55
 
 
91
  from transformers import AutoTokenizer
92
 
93
  tok = AutoTokenizer.from_pretrained("EmpathicRobotics/tokenizer-vla-adaptive-v2")
94
+ print(len(tok)) # 156509
95
 
96
  # All VLA and SNAC tokens are single atomic tokens
97
  print(tok.encode("<seed2_1137>", add_special_tokens=False)) # [59908]
 
122
  # vocab size: 156,505
123
  ```
124
 
125
+ Script: `tools/tokenizer/build_tokenizers.py` in the
126
  [finevideo-vla](https://github.com/TieuDaoChanNhan/finevideo-vla) repo.
127
 
128
  ---
tokenizer.json CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:9d3355c4ef69a8c0a222f2b4bbdccedc4d309454ca9f467c401840f2b35f42c7
3
- size 23746914
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:464e1f6ca5869a2a480bc799df23823b405c579972a73e5063a3d43112f6c43c
3
+ size 23747658