Spaces:
Runtime error
Runtime error
Commit
·
881f140
1
Parent(s):
1b1b2f1
Update pages/gpt_v1.py
Browse files- 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('
|
17 |
|
18 |
|
19 |
col1, col2, col3 = st.columns([5, 2, 12])
|
20 |
|
21 |
with col1:
|
22 |
|
23 |
-
length = st.slider('Длина генерируемой последовательности:', 8, 256,
|
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)
|