From f61f072b610c68fb17f288fec72806d3ce3a5080 Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Wed, 16 Aug 2023 18:11:24 +0200 Subject: [PATCH] Fix `MaskFormerModelIntegrationTest` OOM (#25544) fix Co-authored-by: ydshieh --- tests/models/maskformer/test_modeling_maskformer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/models/maskformer/test_modeling_maskformer.py b/tests/models/maskformer/test_modeling_maskformer.py index 32d5e03b03..a2e3ee306a 100644 --- a/tests/models/maskformer/test_modeling_maskformer.py +++ b/tests/models/maskformer/test_modeling_maskformer.py @@ -539,7 +539,7 @@ class MaskFormerModelIntegrationTest(unittest.TestCase): image_processor = self.default_image_processor inputs = image_processor( - [np.zeros((3, 800, 1333)), np.zeros((3, 800, 1333))], + [np.zeros((3, 400, 333)), np.zeros((3, 400, 333))], segmentation_maps=[np.zeros((384, 384)).astype(np.float32), np.zeros((384, 384)).astype(np.float32)], return_tensors="pt", )