Commit
·
9a23788
1
Parent(s):
07a5a56
Update README.md
Browse files
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
|
17 |
|
18 |
## Usage
|
19 |
|
20 |
-
|
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 |
-
|
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 |
-
|
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
|