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
|