fix decoder_position_ids in decode()
03d8c80
ydshieh
commited on
Commit History
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