orionweller commited on
Commit
9cbd6cd
·
verified ·
1 Parent(s): 3118b6d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -145,7 +145,7 @@ def encode_queries(dataset_name, postfix):
145
  outputs = model(**batch_dict)
146
  embeds = pool(outputs.last_hidden_state, batch_dict['attention_mask'])
147
  embeds = F.normalize(embeds, p=2, dim=-1)
148
- encoded_embeds.append(embeds.cpu().numpy())
149
 
150
  return np.concatenate(encoded_embeds, axis=0)
151
 
 
145
  outputs = model(**batch_dict)
146
  embeds = pool(outputs.last_hidden_state, batch_dict['attention_mask'])
147
  embeds = F.normalize(embeds, p=2, dim=-1)
148
+ encoded_embeds.append(embeds.float().cpu().numpy())
149
 
150
  return np.concatenate(encoded_embeds, axis=0)
151