jmercat commited on
Commit
e2d46d2
·
verified ·
1 Parent(s): b03066e

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +52 -0
README.md ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ library_name: vla-foundry
4
+ tags:
5
+ - foundry
6
+ - vla_foundry
7
+ - llm
8
+ - text-generation
9
+ ---
10
+
11
+ # Foundry-LLM-1.2B-800B
12
+
13
+ A 1.2B parameter language model pretrained on 800B tokens, part of the [VLA Foundry](https://github.com/TRI-ML/vla_foundry) model collection.
14
+
15
+ ## Model Description
16
+
17
+ - **Architecture:** Transformer (24 layers, 2048 hidden dim, 16 heads, SwiGLU FFN, RoPE, QK-norm)
18
+ - **Parameters:** 1.2B (non-embedding)
19
+ - **Tokenizer:** SmolVLM2 (vocab size 49,280)
20
+ - **Training data:** 800B tokens from DCLM-Baseline-1.0
21
+ - **LR schedule:** Warmup + constant (no decay)
22
+ - **Sequence length:** 2048
23
+
24
+ Earlier checkpoint of the Foundry LLM, used as the language backbone for the downstream VLM and VLA models.
25
+
26
+ ## Evaluation Results
27
+
28
+ Multiple-choice reasoning benchmarks:
29
+
30
+ | HellaSwag | MMLU | ARC-e | ARC-c | PIQA | WinoGrande | OpenBookQA | BoolQ |
31
+ |---|---|---|---|---|---|---|---|
32
+ | 64.3 | 26.0 | 70.3 | 37.0 | 75.8 | 60.9 | 40.0 | 63.2 |
33
+
34
+ ## Usage
35
+
36
+ ```bash
37
+ git clone https://github.com/TRI-ML/vla_foundry.git
38
+ cd vla_foundry
39
+ pip install -e .
40
+ ```
41
+
42
+ ```python
43
+ from vla_foundry.models.base_model import BaseModel
44
+ model = BaseModel.from_pretrained("TRI-ML/Foundry-LLM-1.2B-800B")
45
+ ```
46
+
47
+ ## Links
48
+
49
+ - **Project page:** [tri-ml.github.io/vla_foundry](https://tri-ml.github.io/vla_foundry/)
50
+ - **Paper:** [VLA Foundry (arXiv 2604.19728)](https://arxiv.org/abs/2604.19728)
51
+ - **Code:** [github.com/TRI-ML/vla_foundry](https://github.com/TRI-ML/vla_foundry)
52
+ - **Collection:** [VLA Foundry collection](https://huggingface.co/collections/TRI-ML/vla-foundry)