Commit ·
6afcd5d
1
Parent(s): a7d5287
Update README.md
Browse files
README.md
CHANGED
|
@@ -53,12 +53,12 @@ pipe('''Как собрать питон код?''')
|
|
| 53 |
or
|
| 54 |
```python
|
| 55 |
from transformers import GPT2TokenizerFast,GPT2LMHeadModel
|
| 56 |
-
tokenizer = GPT2TokenizerFast.from_pretrained("
|
| 57 |
special_tokens_dict = {'additional_special_tokens': ['<code>', '</code>', '<instructionS>', '<instructionE>', '<next>']}
|
| 58 |
|
| 59 |
tokenizer.add_special_tokens(special_tokens_dict)
|
| 60 |
device = 'cuda:1'
|
| 61 |
-
model = GPT2LMHeadModel.from_pretrained("
|
| 62 |
model.to(device)
|
| 63 |
|
| 64 |
model.resize_token_embeddings(len(tokenizer))
|
|
|
|
| 53 |
or
|
| 54 |
```python
|
| 55 |
from transformers import GPT2TokenizerFast,GPT2LMHeadModel
|
| 56 |
+
tokenizer = GPT2TokenizerFast.from_pretrained("AlexWortega/instruct_rugptMedium")
|
| 57 |
special_tokens_dict = {'additional_special_tokens': ['<code>', '</code>', '<instructionS>', '<instructionE>', '<next>']}
|
| 58 |
|
| 59 |
tokenizer.add_special_tokens(special_tokens_dict)
|
| 60 |
device = 'cuda:1'
|
| 61 |
+
model = GPT2LMHeadModel.from_pretrained("AlexWortega/instruct_rugptMedium")
|
| 62 |
model.to(device)
|
| 63 |
|
| 64 |
model.resize_token_embeddings(len(tokenizer))
|