Skip to content

Commit 137307a

Browse files
committed
Cleanup
1 parent 965f825 commit 137307a

File tree

1 file changed

+0
-18
lines changed

1 file changed

+0
-18
lines changed

tests/models/encoder_decoder/vision_language/test_mllama.py

Lines changed: 0 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -218,24 +218,6 @@ def process(hf_inputs: BatchEncoding):
218218
)
219219

220220

221-
SIZES = [
222-
# Text only
223-
[],
224-
# Single-size
225-
[(512, 512)],
226-
# Single-size, batched
227-
[(512, 512), (512, 512), (512, 512)],
228-
# Multi-size, batched
229-
[(512, 512), (1024, 512), (1536, 512), (2048, 512), (512, 1024),
230-
(1024, 1024), (512, 1536), (512, 2028)],
231-
# Multi-size, batched, including text only
232-
[(512, 512), (1024, 512), (1536, 512), (2048, 512), (512, 1024),
233-
(1024, 1024), (512, 1536), (512, 2028), None],
234-
# mllama has 8 possible aspect ratios, carefully set the sizes
235-
# to cover all of them
236-
]
237-
238-
239221
@large_gpu_test(min_gb=48)
240222
@pytest.mark.parametrize("model", models)
241223
@pytest.mark.parametrize(

0 commit comments

Comments
 (0)