We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
OrtBackend
1 parent 7755758 commit c065defCopy full SHA for c065def
backends/ort/src/lib.rs
@@ -167,8 +167,8 @@ impl Backend for OrtBackend {
167
168
// Run model
169
let outputs = self.session.run(inputs).e()?;
170
- // Get last_hidden_state ndarray
171
+ // Get last_hidden_state ndarray
172
let outputs = outputs
173
.get("last_hidden_state")
174
.or(outputs.get("token_embeddings"))
@@ -362,6 +362,7 @@ impl Backend for OrtBackend {
362
363
364
365
+
366
// Get last_hidden_state ndarray
367
let outputs = outputs["logits"]
368
.try_extract_tensor::<f32>()
0 commit comments