flax-community
/

vit-gpt2

Model card Files Files and versions Metrics Training metrics Community

vit-gpt2 / vit_gpt2

Commit History

clean up config script

ac161f7

ydshieh commited on Aug 2, 2021

change to add_cross_attention

58a121d

ydshieh commited on Aug 1, 2021

Clean Flax ViT + GPT2-LM script

3ed2a5d

ydshieh commited on Aug 1, 2021

Fix GPT2 with cross-attn

155e823

ydshieh commited on Aug 1, 2021

revert to orig flax_gpt2

1ad3ac0

ydshieh commited on Jul 31, 2021

add codes

a244e91

ydshieh commited on Jul 15, 2021