flax-community
/

vit-gpt2

Model card Files Files and versions Metrics Training metrics Community

vit-gpt2 / vit_gpt2 /modeling_flax_gpt2.py

Commit History

fix query split_heads

0817d00

ydshieh commited on Aug 2, 2021

change to add_cross_attention

58a121d

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