Skip to content

GitLab

Unverified Commit f61f072b authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

Fix `MaskFormerModelIntegrationTest` OOM (#25544) 



fix
Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent 0ed23e4d
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/models/maskformer/test_modeling_maskformer.py
View file @ f61f072b
...@@ -539,7 +539,7 @@ class MaskFormerModelIntegrationTest(unittest.TestCase): ...@@ -539,7 +539,7 @@ class MaskFormerModelIntegrationTest(unittest.TestCase):
image_processor = self.default_image_processor image_processor = self.default_image_processor
inputs = 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)], segmentation_maps=[np.zeros((384, 384)).astype(np.float32), np.zeros((384, 384)).astype(np.float32)],
return_tensors="pt", return_tensors="pt",
) )
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment