[CI Failure] Fix Language Models Test (Extended Pooling) daily CI Failure (#36761)

Signed-off-by: wang.yuqi <yuqi.wang@daocloud.io>

[CI Failure] Fix Language Models Test (Extended Pooling) daily CI Failure (#36761)
Signed-off-by: wang.yuqi <yuqi.wang@daocloud.io>
6ecabe49 · wang.yuqi · GitHub · 2f8b4ce0 · 6ecabe49
Unverified Commit 6ecabe49 authored Mar 12, 2026 by wang.yuqi Committed by GitHub Mar 12, 2026
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

tests/models/language/pooling/test_mm_classifier_conversion.py .../models/language/pooling/test_mm_classifier_conversion.py +2 -1

No files found.
--- a/tests/models/language/pooling/test_mm_classifier_conversion.py
+++ b/tests/models/language/pooling/test_mm_classifier_conversion.py
@@ -32,7 +32,8 @@ def test_idefics_multimodal(


 def update_config(config):
-    config.text_config.update(
+    text_config = config.get_text_config()
+    text_config.update(
        {
            "architectures": ["Gemma3ForSequenceClassification"],
            "classifier_from_token": ["A", "B", "C", "D", "E"],