Skip to content

Commit 5d18201

Browse files
authored
Update backends/ort/src/lib.rs
1 parent 071dbf9 commit 5d18201

File tree

1 file changed

+0
-1
lines changed

1 file changed

+0
-1
lines changed

backends/ort/src/lib.rs

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -204,7 +204,6 @@ impl Backend for OrtBackend {
204204
let pooled_embeddings = match self.pool {
205205
// CLS pooling
206206
Pool::Cls => outputs.slice(s![.., 0, ..]).into_owned().into_dyn(),
207-
// Last token pooling is not supported for this model
208207
Pool::LastToken => {
209208
let axis_len = outputs.len_of(Axis(1));
210209
outputs.slice(s![.., axis_len - 1, ..]).into_owned().into_dyn()

0 commit comments

Comments
 (0)