Hans Elias J commited on
Commit ·
ce64906
1
Parent(s): 0996185
fix norm
Browse files- handler.py +2 -2
handler.py
CHANGED
|
@@ -30,7 +30,7 @@ class EndpointHandler():
|
|
| 30 |
|
| 31 |
outputs = self.model(**batch_dict)
|
| 32 |
|
| 33 |
-
embeddings = average_pool(outputs.last_hidden_state, batch_dict['attention_mask'])
|
| 34 |
-
embeddings = F.normalize(embeddings, p=2, dim=1)
|
| 35 |
|
| 36 |
return embeddings
|
|
|
|
| 30 |
|
| 31 |
outputs = self.model(**batch_dict)
|
| 32 |
|
| 33 |
+
embeddings = average_pool(outputs.last_hidden_state, batch_dict['attention_mask'])
|
| 34 |
+
embeddings = F.normalize(embeddings, p=2, dim=1).tolist()
|
| 35 |
|
| 36 |
return embeddings
|