Skip to content

Commit 47926b7

Browse files
committed
add cohere test
Signed-off-by: Kyle Sayers <[email protected]>
1 parent 59c43fa commit 47926b7

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

tests/llmcompressor/transformers/tracing/test_models.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@
1212
Qwen2_5_VLForConditionalGeneration,
1313
Qwen2VLForConditionalGeneration,
1414
WhisperForConditionalGeneration,
15+
Cohere2VisionForConditionalGeneration,
1516
)
1617

1718
from llmcompressor.pipelines.sequential.helpers import match_modules
@@ -87,6 +88,13 @@
8788
"vision",
8889
["torchvision"],
8990
),
91+
(
92+
"CohereLabs/command-a-vision-07-2025",
93+
Cohere2VisionForConditionalGeneration,
94+
["Cohere2DecoderLayer"],
95+
"vision",
96+
[],
97+
)
9098
(
9199
"Qwen/Qwen2-VL-2B-Instruct",
92100
Qwen2VLForConditionalGeneration,

0 commit comments

Comments
 (0)