Skip to content

Commit a9c3c57

Browse files
committed
v1.1.6 update some desc
1 parent 2125f62 commit a9c3c57

File tree

1 file changed

+7
-5
lines changed

1 file changed

+7
-5
lines changed

README.md

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -29,8 +29,11 @@
2929
</div>
3030
<br>
3131

32-
> [!NOTE]
33-
> 🚧 This repository is fully open source on arXiv, Github and HuggingFace. Welcome to star and cite our work! ✨
32+
<div align="center" style="display: flex; gap: 5px; justify-content: center;">
33+
<p>
34+
<b>🚧 Our workis fully open source on arXiv, Github and HuggingFace. Welcome to star and cite our work! ✨</b>
35+
</p>
36+
</div>
3437

3538
## 📖 Overview
3639

@@ -104,6 +107,7 @@ SQL-R1/
104107
## 🛠️ Environment Setup
105108

106109
> [!NOTE]
110+
> Before getting started, make sure your computing environment supports the following settings:
107111
> - Environment: Python 3.9+
108112
> - CUDA Version: 12.0+ (for verl and vllm integration)
109113
> - GPU Prerequisites: 8 x 80GB+ GPU (for training) / 2 x 40GB GPU (for inference)
@@ -132,11 +136,9 @@ pip install wandb IPython matplotlib sqlparse func_timeout
132136

133137
4. Download the model weights from HuggingFace and put them in the `models/` directory.
134138

135-
5. For training, copy the training dataset in the `example_data` directory to the `data` directory. For inference, copy the database information in the `db_info` directory to the related dataset (`data/Spider`, `data/BIRD`) directory.
139+
5. For training, copy the training dataset in the `example_data` directory to the `data` directory. For inference, copy the database information in the `db_info` directory (including files forSpider-dev, Spider-test and BIRD-dev) to the related dataset (`data/Spider`, `data/BIRD`) directory.
136140

137141
## 🚀 Quick Start
138-
> [!NOTE]
139-
> Please set the related data paths and params before running the scripts.
140142

141143
1. Run training:
142144
```bash

0 commit comments

Comments
 (0)