AlexWortega commited on
Commit
6afcd5d
1 Parent(s): a7d5287

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -53,12 +53,12 @@ pipe('''Как собрать питон код?''')
53
  or
54
  ```python
55
  from transformers import GPT2TokenizerFast,GPT2LMHeadModel
56
- tokenizer = GPT2TokenizerFast.from_pretrained("sberbank-ai/rugpt3small_based_on_gpt2")
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("ckpts11/sft_0_70000/")
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))