Spaces:

Ilvir
/

ilva

Runtime error

Ilvir commited on Jun 9, 2023

Commit

d37aaf6

1 Parent(s): e037f46

Update pages/gpt.py

Files changed (1) hide show

pages/gpt.py CHANGED Viewed

@@ -14,7 +14,7 @@ model = GPT2LMHeadModel.from_pretrained(
     output_hidden_states = False,
 )
-model.load_state_dict(torch.load('models/model.pt', map_location=torch.device('cpu')))
 # length = st.sidebar.slider('**Generated sequence length:**', 8, 256, 15)

     output_hidden_states = False,
 )
+model.load_state_dict(torch.load('models/modelgpt.pt', map_location=torch.device('cpu')))
 # length = st.sidebar.slider('**Generated sequence length:**', 8, 256, 15)