Skip to content

GitLab

Unverified Commit 6ecabe49 authored by wang.yuqi's avatar wang.yuqi Committed by GitHub
Browse files

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


Signed-off-by: default avatarwang.yuqi <yuqi.wang@daocloud.io>
parent 2f8b4ce0
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
tests/models/language/pooling/test_mm_classifier_conversion.py
View file @ 6ecabe49
...@@ -32,7 +32,8 @@ def test_idefics_multimodal( ...@@ -32,7 +32,8 @@ def test_idefics_multimodal(
def update_config(config): def update_config(config):
config.text_config.update( text_config = config.get_text_config()
text_config.update(
{ {
"architectures": ["Gemma3ForSequenceClassification"], "architectures": ["Gemma3ForSequenceClassification"],
"classifier_from_token": ["A", "B", "C", "D", "E"], "classifier_from_token": ["A", "B", "C", "D", "E"],
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment