From e1c72fd41c6254bc40da140c113796f811d6a9bb Mon Sep 17 00:00:00 2001
From: flybird11111 <1829166702@qq.com>
Date: Thu, 22 May 2025 16:49:06 +0800
Subject: [PATCH] fix

---
 tests/kit/model_zoo/transformers/bert.py | 1 +
 1 file changed, 1 insertion(+)

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