Fix style
64afcd5
ydshieh
commited on
Commit History
Change a parameter in decode(): deterministic -> train
7d3b1a0
ydshieh
commited on
final ver
87485e5
ydshieh
commited on
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
b8c22f0
ydshieh
commited on
Fix project_encoder
e30ab96
ydshieh
commited on
Fix project_encoder
5081c5d
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 pixel_values being casted to int
89ca80d
ydshieh
commited on
fix decoder_position_ids in decode()
03d8c80
ydshieh
commited on
fix pad token id issue
9aceda3
ydshieh
commited on
fix attention_mask missing
4bf5c74
ydshieh
commited on
fix FlaxViTGPT2LMModule return value
bfef308
ydshieh
commited on
fix query split_heads
0817d00
ydshieh
commited on
fix model args
b88266d
ydshieh
commited on
update model script
199889f
ydshieh
commited on
update model script
b00cdfe
ydshieh
commited on
update config script
b31314b
ydshieh
commited on
update config script
b04e4c6
ydshieh
commited on
clean up config script
ac161f7
ydshieh
commited on
change to add_cross_attention
58a121d
ydshieh
commited on
Clean Flax ViT + GPT2-LM script
3ed2a5d
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