shibing624 commited on
Commit
9a23788
·
1 Parent(s): 07a5a56

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +7 -7
README.md CHANGED
@@ -13,11 +13,11 @@ license: "apache-2.0"
13
  # GPT2 for Code AutoComplete Model
14
  code-autocomplete, a code completion plugin for Python.
15
 
16
- **code-autocomplete**实现了Python代码行粒度和块粒度自动补全功能。
17
 
18
  ## Usage
19
 
20
- 本项目开源在中文文本纠错项目:[code-autocomplete](https://github.com/shibing624/code-autocomplete),可支持GPT2模型,通过如下命令调用:
21
 
22
  ```python
23
  from autocomplete.gpt2 import Infer
@@ -26,7 +26,7 @@ i = m.predict('import torch.nn as')
26
  print(i)
27
  ```
28
 
29
- 当然,你也可使用官方的huggingface/transformers调用:
30
 
31
  *Please use 'GPT2' related functions to load this model!*
32
 
@@ -111,7 +111,7 @@ Process finished with exit code 0
111
 
112
  ```
113
 
114
- 模型文件组成:
115
  ```
116
  code-autocomplete-gpt2-base
117
  ├── config.json
@@ -122,8 +122,8 @@ code-autocomplete-gpt2-base
122
  └── vocab.json
123
  ```
124
 
125
- ### 训练数据集
126
- #### pytorch_awesome的所有项目代码
127
 
128
  download [code-autocomplete](https://github.com/shibing624/code-autocomplete),
129
  ```shell
@@ -131,7 +131,7 @@ cd autocomplete
131
  python create_dataset.py
132
  ```
133
 
134
- 如果需要训练code-autocomplete,请参考[https://github.com/shibing624/code-autocomplete/blob/main/autocomplete/gpt2.py](https://github.com/shibing624/code-autocomplete/blob/main/autocomplete/gpt2.py)
135
 
136
 
137
  ### About GPT2
 
13
  # GPT2 for Code AutoComplete Model
14
  code-autocomplete, a code completion plugin for Python.
15
 
16
+ **code-autocomplete** can Automatic completion of code line granularity and block granularity.
17
 
18
  ## Usage
19
 
20
+ Open source repo:[code-autocomplete](https://github.com/shibing624/code-autocomplete),support GPT2 model, usage:
21
 
22
  ```python
23
  from autocomplete.gpt2 import Infer
 
26
  print(i)
27
  ```
28
 
29
+ Also, use huggingface/transformers
30
 
31
  *Please use 'GPT2' related functions to load this model!*
32
 
 
111
 
112
  ```
113
 
114
+ Model files:
115
  ```
116
  code-autocomplete-gpt2-base
117
  ├── config.json
 
122
  └── vocab.json
123
  ```
124
 
125
+ ### Train data
126
+ #### pytorch_awesome projects source code
127
 
128
  download [code-autocomplete](https://github.com/shibing624/code-autocomplete),
129
  ```shell
 
131
  python create_dataset.py
132
  ```
133
 
134
+ If you want train code-autocomplete GPT2 model,refer [https://github.com/shibing624/code-autocomplete/blob/main/autocomplete/gpt2.py](https://github.com/shibing624/code-autocomplete/blob/main/autocomplete/gpt2.py)
135
 
136
 
137
  ### About GPT2