jmercat commited on
Commit
8ef8731
·
verified ·
1 Parent(s): 61938df

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
+ - vlm
8
+ - image-text-to-text
9
+ ---
10
+
11
+ # Foundry-VLM-1.3B-200M
12
+
13
+ A 1.3B parameter vision-language model trained on 200M image-caption samples, part of the [VLA Foundry](https://github.com/TRI-ML/vla_foundry) collection.
14
+
15
+ ## Model Description
16
+
17
+ - **Architecture:** ViT encoder (12 layers, 768 hidden dim, patch size 14, pixel-shuffle 2x) + Transformer decoder (24 layers, 2048 hidden dim, 16 heads)
18
+ - **Parameters:** 1.3B (non-embedding)
19
+ - **Processor:** SmolVLM2
20
+ - **Training data:** 200M image-caption pairs from DataComp-DR-1B
21
+ - **LR schedule:** Warmup + constant for 165M samples, then 35M samples of cosine decay
22
+ - **LLM backbone:** Initialized from [Foundry-LLM-1.2B-800B](https://huggingface.co/TRI-ML/Foundry-LLM-1.2B-800B)
23
+
24
+ Continuation of [Foundry-VLM-1.3B-165M](https://huggingface.co/TRI-ML/Foundry-VLM-1.3B-165M) with an additional 35M samples of cosine-decayed training. Used as the vision-language backbone for the Foundry-VLA-1.7B action models.
25
+
26
+ ## Evaluation Results
27
+
28
+ COCO-val captioning:
29
+
30
+ | BLEU-1 | BLEU-2 | BLEU-3 | BLEU-4 | ROUGE-L | CIDEr |
31
+ |---|---|---|---|---|---|
32
+ | 58.64 | 38.62 | 24.49 | 15.57 | 38.17 | 55.14 |
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-VLM-1.3B-200M")
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)