SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Fix MaskFormerModelIntegrationTest OOM (#25544)

Browse Source 
fix

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar
2023-08-16 18:11:24 +02:00
committed by GitHub
parent 0ed23e4db2
commit f61f072b61
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/models/maskformer/test_modeling_maskformer.py
View File

@@ -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",
) )