Spaces:

vasevooo
/

NLP_project

Sleeping

vasevooo commited on Jun 8, 2023

Commit

05b2427

1 Parent(s): 3521368

Update pages/gpt.py

Files changed (1) hide show

pages/gpt.py CHANGED Viewed

@@ -7,7 +7,7 @@ model = GPT2LMHeadModel.from_pretrained(
     output_hidden_states = False,
 )
 # Вешаем сохраненные веса на нашу модель
-model.load_state_dict(torch.load('models/model.pt'), map_location=torch.device('cpu'))
 def generate_text(model, tokenizer, prompt, length, num_samples, temperature):

     output_hidden_states = False,
 )
 # Вешаем сохраненные веса на нашу модель
+model.load_state_dict(torch.load('models/model.pt', map_location=torch.device('cpu')))
 def generate_text(model, tokenizer, prompt, length, num_samples, temperature):