vasevooo commited on
Commit
eebac25
·
1 Parent(s): 7651de7

Update pages/gpt.py

Browse files
Files changed (1) hide show
  1. pages/gpt.py +4 -0
pages/gpt.py CHANGED
@@ -7,6 +7,10 @@ model = GPT2LMHeadModel.from_pretrained(
7
  output_attentions = False,
8
  output_hidden_states = False,
9
  )
 
 
 
 
10
  # Вешаем сохраненные веса на нашу модель
11
  model.load_state_dict(torch.load('models/model.pt', map_location=torch.device('cpu')))
12
 
 
7
  output_attentions = False,
8
  output_hidden_states = False,
9
  )
10
+
11
+ tokenizer = GPT2Tokenizer.from_pretrained('sberbank-ai/rugpt3small_based_on_gpt2')
12
+
13
+
14
  # Вешаем сохраненные веса на нашу модель
15
  model.load_state_dict(torch.load('models/model.pt', map_location=torch.device('cpu')))
16