refactor: deprecated method

kozistr · Nov 17, 2024 · 2fbf552 · 2fbf552
1 parent 8f361db
commit 2fbf552
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/backends/ort/src/lib.rs b/backends/ort/src/lib.rs
@@ -246,7 +246,7 @@ impl Backend for OrtBackend {
         if has_raw_requests {
             // Reshape outputs
             let s = outputs.shape().to_vec();
-            let outputs = outputs.into_shape((s[0] * s[1], s[2])).e()?;
+            let outputs = outputs.into_shape_with_order((s[0] * s[1], s[2])).e()?;
 
             // We need to remove the padding tokens only if batch_size > 1 and there are some
             // member of the batch that require pooling