Update pixel_attention_mask unit test

huggingface · Dec 2, 2024 · f88500c · f88500c
1 parent 45de96a
commit f88500c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/processors.test.js b/tests/processors.test.js
@@ -701,7 +701,7 @@ describe("Processors", () => {
           );
           compare(pixel_attention_mask.dims, [3, 2, 364, 364]);
           compare(
-            pixel_attention_mask.mean().item(),
+            pixel_attention_mask.to('float32').mean().item(),
             2 / 3,
             0.001, // threshold
           );