[CI slow] Fix expected values (#27999)

* fix expected values * style * test is slow
huggingface · Dec 13, 2023 · ec43d68 · ec43d68
1 parent 749f94e
commit ec43d68
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/models/vit_msn/test_modeling_vit_msn.py b/tests/models/vit_msn/test_modeling_vit_msn.py
@@ -227,6 +227,6 @@ def test_inference_image_classification_head(self):
         expected_shape = torch.Size((1, 1000))
         self.assertEqual(outputs.logits.shape, expected_shape)
 
-        expected_slice = torch.tensor([-0.0803, -0.4454, -0.2375]).to(torch_device)
+        expected_slice = torch.tensor([0.5588, 0.6853, -0.5929]).to(torch_device)
 
         self.assertTrue(torch.allclose(outputs.logits[0, :3], expected_slice, atol=1e-4))