Update README.md
Browse files
README.md
CHANGED
@@ -12,9 +12,9 @@ tasks:
|
|
12 |
|
13 |
# Model Card for NL2SQL-StarCoder-15B
|
14 |
|
15 |
-
##
|
16 |
|
17 |
-
NL2SQL-StarCoder-15B
|
18 |
|
19 |
## Requirements
|
20 |
|
@@ -22,9 +22,9 @@ NL2SQL-StarCoder-15B 是在基础模型 StarCoder 上通过 QLoRA 对自然语
|
|
22 |
- pytorch>=2.0.0
|
23 |
- transformers==4.32.0
|
24 |
- CUDA 11.4
|
25 |
-
##
|
26 |
|
27 |
-
|
28 |
```python
|
29 |
"""
|
30 |
<|user|>
|
@@ -35,15 +35,15 @@ CREATE TABLE "table_name" (
|
|
35 |
...
|
36 |
)
|
37 |
|
38 |
-
/* Write a sql to answer the following question: {
|
39 |
<|assistant|>
|
40 |
```sql
|
41 |
-
{
|
42 |
```<|end|>
|
43 |
"""
|
44 |
```
|
45 |
|
46 |
-
##
|
47 |
|
48 |
```python
|
49 |
import torch
|
|
|
12 |
|
13 |
# Model Card for NL2SQL-StarCoder-15B
|
14 |
|
15 |
+
## Model Inro
|
16 |
|
17 |
+
NL2SQL-StarCoder-15B is a NLP-SQL model fintuned by QLoRAbased on StarCoder 15B Code-LLM。
|
18 |
|
19 |
## Requirements
|
20 |
|
|
|
22 |
- pytorch>=2.0.0
|
23 |
- transformers==4.32.0
|
24 |
- CUDA 11.4
|
25 |
+
## Data Format
|
26 |
|
27 |
+
The data is in the form of a string spliced by the model in the training data format, which is also how the input PROMPT is spliced during inference:
|
28 |
```python
|
29 |
"""
|
30 |
<|user|>
|
|
|
35 |
...
|
36 |
)
|
37 |
|
38 |
+
/* Write a sql to answer the following question: {Question} */
|
39 |
<|assistant|>
|
40 |
```sql
|
41 |
+
{Output SQL}
|
42 |
```<|end|>
|
43 |
"""
|
44 |
```
|
45 |
|
46 |
+
## Quick Start
|
47 |
|
48 |
```python
|
49 |
import torch
|