diff options
Diffstat (limited to 'candle-wasm-examples/segment-anything/src')
| -rw-r--r-- | candle-wasm-examples/segment-anything/src/bin/m.rs | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/candle-wasm-examples/segment-anything/src/bin/m.rs b/candle-wasm-examples/segment-anything/src/bin/m.rs index acd903b0..a56e48c2 100644 --- a/candle-wasm-examples/segment-anything/src/bin/m.rs +++ b/candle-wasm-examples/segment-anything/src/bin/m.rs @@ -94,7 +94,7 @@ impl Model { &embeddings.data, embeddings.height as usize, embeddings.width as usize, - Some((x, y)), + &[(x, y)], false, )?; let iou = iou_predictions.flatten(0, 1)?.to_vec1::<f32>()?[0]; |