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
|