We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 59c43fa commit 47926b7Copy full SHA for 47926b7
tests/llmcompressor/transformers/tracing/test_models.py
@@ -12,6 +12,7 @@
12
Qwen2_5_VLForConditionalGeneration,
13
Qwen2VLForConditionalGeneration,
14
WhisperForConditionalGeneration,
15
+ Cohere2VisionForConditionalGeneration,
16
)
17
18
from llmcompressor.pipelines.sequential.helpers import match_modules
@@ -87,6 +88,13 @@
87
88
"vision",
89
["torchvision"],
90
),
91
+ (
92
+ "CohereLabs/command-a-vision-07-2025",
93
94
+ ["Cohere2DecoderLayer"],
95
+ "vision",
96
+ [],
97
+ )
98
(
99
"Qwen/Qwen2-VL-2B-Instruct",
100
0 commit comments