From 6ecabe493628eeeafc346dbe2ee3b6c0a6d94a80 Mon Sep 17 00:00:00 2001 From: "wang.yuqi" Date: Thu, 12 Mar 2026 12:22:05 +0800 Subject: [PATCH] [CI Failure] Fix Language Models Test (Extended Pooling) daily CI Failure (#36761) Signed-off-by: wang.yuqi --- tests/models/language/pooling/test_mm_classifier_conversion.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/models/language/pooling/test_mm_classifier_conversion.py b/tests/models/language/pooling/test_mm_classifier_conversion.py index 78448de594..5ad48905b1 100644 --- 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"],