ligeng-zhu-nv commited on
Commit
d2520a6
·
verified ·
1 Parent(s): fc1cd6e

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +99 -0
README.md ADDED
@@ -0,0 +1,99 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ language:
4
+ - en
5
+ library_name: transformers
6
+ base_model:
7
+ - Qwen/Qwen3-VL-8B-Thinking
8
+ pipeline_tag: image-text-to-text
9
+ tags:
10
+ - visual-grounding
11
+ - multimodal
12
+ - qwen3-vl
13
+ - supervised-fine-tuning
14
+ ---
15
+
16
+ # EGM-Qwen3-VL-8B-SFT
17
+
18
+ <p align="center">
19
+ <a href="https://nvlabs.github.io/EGM">[Project Page]</a> &nbsp;
20
+ <a href="https://github.com/NVlabs/EGM">[Code]</a> &nbsp;
21
+ </p>
22
+
23
+ <div align="center">
24
+ <img src="https://nvlabs.github.io/EGM/figure4.jpeg" width="90%"/>
25
+ </div>
26
+
27
+ ## Model Summary
28
+
29
+ **EGM-Qwen3-VL-8B-SFT** is the supervised fine-tuning (SFT) checkpoint from the first stage of the [EGM (Efficient Visual Grounding Language Models)](https://nvlabs.github.io/EGM) training pipeline. It is built on top of [Qwen3-VL-8B-Thinking](https://huggingface.co/Qwen/Qwen3-VL-8B-Thinking).
30
+
31
+ This is an **intermediate checkpoint** intended for further reinforcement learning training. For the final model with best performance, see [nvidia/EGM-8B](https://huggingface.co/nvidia/EGM-8B).
32
+
33
+ ## Training Details
34
+
35
+ ### SFT Stage
36
+
37
+ In the SFT stage, a proprietary VLM generates detailed chain-of-thought reasoning steps for visual grounding training data. The base Qwen3-VL-8B-Thinking model is then fine-tuned on this reasoning-augmented data to learn structured visual grounding with explicit reasoning.
38
+
39
+ This SFT checkpoint serves as the initialization for the subsequent RL stage (GRPO), which yields the final [EGM-8B](https://huggingface.co/nvidia/EGM-8B) model.
40
+
41
+ ### How to Use for RL Training
42
+
43
+ ```bash
44
+ pip install -U huggingface_hub
45
+ huggingface-cli download nvidia/EGM-8B-SFT --local-dir ./models/EGM-8B-SFT
46
+ ```
47
+
48
+ Then follow the installation instructions in the [EGM repository](https://github.com/NVlabs/EGM#installation), prepare the RL data and start training:
49
+
50
+ ```bash
51
+ export BASE_DIR=$(pwd)
52
+ export MODEL_PATH="${BASE_DIR}/models/EGM-8B-SFT"
53
+ export OUTPUT_DIR="${BASE_DIR}/checkpoint/"
54
+ export DATA_DIR="${BASE_DIR}/data/EGM_Datasets/processed_rl_data/"
55
+
56
+ cd verl
57
+ bash scripts/grounding_qwen.sh
58
+ ```
59
+
60
+ See the [EGM repository](https://github.com/NVlabs/EGM#rl-training) for full RL training instructions.
61
+
62
+ ## Model Architecture
63
+
64
+ | Component | Details |
65
+ |---|---|
66
+ | Architecture | Qwen3VLForConditionalGeneration |
67
+ | Precision | bfloat16 |
68
+ | Text Hidden Size | 4096 |
69
+ | Text Layers | 36 |
70
+ | Attention Heads | 32 (8 KV heads) |
71
+ | Text Intermediate Size | 12,288 |
72
+ | Vision Hidden Size | 1152 |
73
+ | Vision Layers | 27 |
74
+ | Patch Size | 16 x 16 |
75
+ | Max Position Embeddings | 262,144 |
76
+ | Vocabulary Size | 151,936 |
77
+
78
+ ## Related Models
79
+
80
+ | Model | Description |
81
+ |---|---|
82
+ | [nvidia/EGM-8B](https://huggingface.co/nvidia/EGM-8B) | Final RL-trained model (best performance) |
83
+ | [nvidia/EGM-4B-SFT](https://huggingface.co/nvidia/EGM-4B-SFT) | SFT checkpoint for the 4B variant |
84
+ | [nvidia/EGM-4B](https://huggingface.co/nvidia/EGM-4B) | Final RL-trained 4B model |
85
+
86
+ ## Citation
87
+
88
+ ```bibtex
89
+ @article{zhan2026EGM,
90
+ author = {Zhan, Guanqi and Li, Changye and Liu, Zhijian and Lu, Yao and Wu, Yi and Han, Song and Zhu, Ligeng},
91
+ title = {EGM: Efficient Visual Grounding Language Models},
92
+ booktitle = {arXiv},
93
+ year = {2026}
94
+ }
95
+ ```
96
+
97
+ ## Acknowledgment
98
+
99
+ This repository benefits from [Qwen3-VL](https://github.com/QwenLM/Qwen3-VL), [InternVL](https://github.com/OpenGVLab/InternVL), [verl](https://github.com/volcengine/verl) and [verl-internvl](https://github.com/Weiyun1025/verl-internvl).