fix

2025-06-23 22:19:47 +00:00 · 2025-05-22 16:49:06 +08:00 · 2025-05-22 16:49:06 +08:00 · e1c72fd41c
commit e1c72fd41c
parent bafc80c3b0
1 changed files with 1 additions and 0 deletions
--- 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