Skip to content

Commit b791f87

Browse files
committed
Add assertion for vision_use_head in Gemma3VLBridge unit tests
Signed-off-by: Ao Tang <aot@nvidia.com>
1 parent e2e5787 commit b791f87

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

‎tests/unit_tests/models/gemma_vl/test_gemma3_vl_bridge.py‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -166,6 +166,7 @@ def test_provider_bridge_vl_specific_config(self, gemma3_vl_bridge, mock_hf_pret
166166

167167
# Check vision config
168168
assert isinstance(provider.vision_config, SiglipVisionConfig)
169+
assert provider.vision_config.vision_use_head is False
169170
assert provider.mm_tokens_per_image == 256
170171

171172
def test_provider_bridge_vision_projector_config(self, gemma3_vl_bridge, mock_hf_pretrained):

0 commit comments

Comments
 (0)