Skip to content

Commit c065def

Browse files
committed
Fix spacing in OrtBackend
1 parent 7755758 commit c065def

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

backends/ort/src/lib.rs

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -167,8 +167,8 @@ impl Backend for OrtBackend {
167167

168168
// Run model
169169
let outputs = self.session.run(inputs).e()?;
170-
// Get last_hidden_state ndarray
171170

171+
// Get last_hidden_state ndarray
172172
let outputs = outputs
173173
.get("last_hidden_state")
174174
.or(outputs.get("token_embeddings"))
@@ -362,6 +362,7 @@ impl Backend for OrtBackend {
362362

363363
// Run model
364364
let outputs = self.session.run(inputs).e()?;
365+
365366
// Get last_hidden_state ndarray
366367
let outputs = outputs["logits"]
367368
.try_extract_tensor::<f32>()

0 commit comments

Comments
 (0)