final ver
87485e5
ydshieh
commited on
Commit History
revert
3a17811
ydshieh
commited on
no residual for proj
67d7ae6
ydshieh
commited on
fix output types
0f3d46c
ydshieh
commited on
make more similar to pytorch gpt2 code
166db12
ydshieh
commited on
revert to original flax gpt2
7f266a7
ydshieh
commited on
Change check to crossattention
5a95e3a
ydshieh
commited on
Change Flax GPT2 with cross-attn layers to be the same as PyTorch's version
7d29388
ydshieh
commited on
Change Flax GPT2 with cross-attn outputs to be the same as PyTorch's version
165ad1e
ydshieh
commited on
remove only_self_attn
a01b02a
ydshieh
commited on
Fix project_encoder
9a48a1a
ydshieh
commited on
Fix project_encoder
7262cfd
ydshieh
commited on
Add project_encoder and related layers
54ece9e
ydshieh
commited on
Make a remark about a serious bug I made previously
0ac6b6e
ydshieh
commited on
fix query split_heads
0817d00
ydshieh
commited on
change to add_cross_attention
58a121d
ydshieh
commited on
Fix GPT2 with cross-attn
155e823
ydshieh
commited on
revert to orig flax_gpt2
1ad3ac0
ydshieh
commited on
add codes
a244e91
ydshieh
commited on