AlexWortega
commited on
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))
|