diff --git a/tests/unit/conftest.py b/tests/unit/conftest.py
index 18bece9aa9d..68f684a7c28 100644
--- a/tests/unit/conftest.py
+++ b/tests/unit/conftest.py
@@ -652,6 +652,7 @@ def tiny_qwen2_model_path():
         vocab_size=151936,
         tie_word_embeddings=False,
         num_key_value_heads=None,
+        classifier_dropout=0.0,
     )
     model = Qwen2ForCausalLM(config=config)
     tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen2-1.5B")