mirror of
https://github.com/huggingface/candle.git
synced 2025-06-15 10:26:33 +00:00
Add negative prompts to segment-anything. (#1000)
This commit is contained in:
@ -94,7 +94,7 @@ impl Model {
|
||||
&embeddings.data,
|
||||
embeddings.height as usize,
|
||||
embeddings.width as usize,
|
||||
&[(x, y)],
|
||||
&[(x, y, true)],
|
||||
false,
|
||||
)?;
|
||||
let iou = iou_predictions.flatten(0, 1)?.to_vec1::<f32>()?[0];
|
||||
|
Reference in New Issue
Block a user