diff --git a/tests/kit/model_zoo/transformers/bert.py b/tests/kit/model_zoo/transformers/bert.py
index 6dd3e102c..03c8ac201 100644
--- a/tests/kit/model_zoo/transformers/bert.py
+++ b/tests/kit/model_zoo/transformers/bert.py
@@ -370,6 +370,7 @@ config = transformers.BertConfig(
     intermediate_size=256,
     hidden_dropout_prob=0,
     attention_probs_dropout_prob=0,
+    attn_implementation="eager",
 )
 
 # register the BERT variants