MARI-posa commited on
Commit
736ed2e
1 Parent(s): c892af5

Update stri.py

Browse files
Files changed (1) hide show
  1. stri.py +0 -1
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