Spaces:
Runtime error
Runtime error
Update stri.py
Browse files
stri.py
CHANGED
@@ -59,7 +59,6 @@ if st.button('Сгенерировать'):
|
|
59 |
with torch.no_grad():
|
60 |
query_outputs = model(**query_tokens)
|
61 |
query_hidden_states = query_outputs.hidden_states[-1][:,0,:]
|
62 |
-
query_hidden_states = torch.nn.functional.normalize(query_hidden_states)query_hidden_states = query_outputs.hidden_states[-1][:,0,:]
|
63 |
query_hidden_states = torch.nn.functional.normalize(query_hidden_states)
|
64 |
|
65 |
|
|
|
59 |
with torch.no_grad():
|
60 |
query_outputs = model(**query_tokens)
|
61 |
query_hidden_states = query_outputs.hidden_states[-1][:,0,:]
|
|
|
62 |
query_hidden_states = torch.nn.functional.normalize(query_hidden_states)
|
63 |
|
64 |
|