Skip to content

Commit 9423e8a

Browse files
committed
fix format
Signed-off-by: Kyle Sayers <[email protected]>
1 parent 47926b7 commit 9423e8a

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

tests/llmcompressor/transformers/tracing/test_models.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@
33
import pytest
44
from transformers import (
55
AutoModelForCausalLM,
6+
Cohere2VisionForConditionalGeneration,
67
Gemma3ForConditionalGeneration,
78
Idefics3ForConditionalGeneration,
89
Llama4ForConditionalGeneration,
@@ -12,7 +13,6 @@
1213
Qwen2_5_VLForConditionalGeneration,
1314
Qwen2VLForConditionalGeneration,
1415
WhisperForConditionalGeneration,
15-
Cohere2VisionForConditionalGeneration,
1616
)
1717

1818
from llmcompressor.pipelines.sequential.helpers import match_modules
@@ -94,8 +94,7 @@
9494
["Cohere2DecoderLayer"],
9595
"vision",
9696
[],
97-
)
98-
(
97+
)(
9998
"Qwen/Qwen2-VL-2B-Instruct",
10099
Qwen2VLForConditionalGeneration,
101100
["Qwen2VLDecoderLayer"],

0 commit comments

Comments
 (0)