gabrielpondc commited on
Commit
1991883
·
verified ·
1 Parent(s): 6644ecd

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +7 -7
README.md CHANGED
@@ -12,9 +12,9 @@ tasks:
12
 
13
  # Model Card for NL2SQL-StarCoder-15B
14
 
15
- ## 模型介绍
16
 
17
- NL2SQL-StarCoder-15B 是在基础模型 StarCoder 上通过 QLoRA 对自然语言生成SQL任务进行微调的 15B Code-LLM。
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
- 推理数据为模型在训练数据格式下拼接的字符串形式,它也是推理时输入prompt拼接的方式:
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
- {输出SQL}
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