github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-06-25 23:11:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
flybird11111 2025-05-22 16:49:06 +08:00
parent bafc80c3b0
commit e1c72fd41c
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tests/kit/model_zoo/transformers/bert.py
View File

@ -370,6 +370,7 @@ config = transformers.BertConfig(
intermediate_size=256, intermediate_size=256,
hidden_dropout_prob=0, hidden_dropout_prob=0,
attention_probs_dropout_prob=0, attention_probs_dropout_prob=0,
attn_implementation="eager",
) )
# register the BERT variants # register the BERT variants