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

Signed-off-by: wang.yuqi <yuqi.wang@daocloud.io>
2026-03-12 12:22:05 +08:00
parent 2f8b4ce0c0
commit 6ecabe4936
1 changed files with 2 additions and 1 deletions
--- 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"],