Adjust outputs (#16)

2025-12-03 02:58:22 +00:00 · 2024-06-01 16:58:17 +08:00
parent a65e2aff7b
commit 20577e4242
9 changed files with 40 additions and 28 deletions
--- a/examples/clip/main.rs
+++ b/examples/clip/main.rs
@@ -39,7 +39,10 @@ fn main() -> Result<(), Box<dyn std::error::Error>> {
        let feats_image = model.encode_images(&images).unwrap();

        // use image to query texts
-        let matrix = feats_image.dot2(&feats_text)?;
+        let matrix = match feats_image.embedding() {
+            Some(x) => x.dot2(feats_text.embedding().unwrap())?,
+            None => continue,
+        };

        // summary
        for i in 0..paths.len() {