vvv-knyazeva commited on
Commit
881f140
·
1 Parent(s): 1b1b2f1

Update pages/gpt_v1.py

Browse files
Files changed (1) hide show
  1. pages/gpt_v1.py +2 -2
pages/gpt_v1.py CHANGED
@@ -13,14 +13,14 @@ model = GPT2LMHeadModel.from_pretrained(
13
  output_hidden_states = False,
14
  )
15
  # Вешаем сохраненные веса на нашу модель
16
- model.load_state_dict(torch.load('modelgpt.pt', map_location=torch.device('cpu')))
17
 
18
 
19
  col1, col2, col3 = st.columns([5, 2, 12])
20
 
21
  with col1:
22
 
23
- length = st.slider('Длина генерируемой последовательности:', 8, 256, 15)
24
  num_samples = st.slider('Число генераций:', 1, 10, 1)
25
  temperature = st.slider('Температура:', 1.0, 10.0, 2.0)
26
  top_k = st.slider('Количество наиболее вероятных слов генерации:', 10, 200, 50)
 
13
  output_hidden_states = False,
14
  )
15
  # Вешаем сохраненные веса на нашу модель
16
+ model.load_state_dict(torch.load('model.pt', map_location=torch.device('cpu')))
17
 
18
 
19
  col1, col2, col3 = st.columns([5, 2, 12])
20
 
21
  with col1:
22
 
23
+ length = st.slider('Длина генерируемой последовательности:', 8, 256, 16)
24
  num_samples = st.slider('Число генераций:', 1, 10, 1)
25
  temperature = st.slider('Температура:', 1.0, 10.0, 2.0)
26
  top_k = st.slider('Количество наиболее вероятных слов генерации:', 10, 200, 50)