vit-gpt2 / vit_gpt2 /modeling_flax_vit_gpt2_lm.py

Commit History

Change a parameter in decode(): deterministic -> train
7d3b1a0

ydshieh commited on

Fix project_encoder
b8c22f0

ydshieh commited on

Fix project_encoder
e30ab96

ydshieh commited on

Add project_encoder and related layers
54ece9e

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 attention_mask missing
4bf5c74

ydshieh commited on

fix FlaxViTGPT2LMModule return value
bfef308

ydshieh commited on

fix model args
b88266d

ydshieh commited on

update model script
199889f

ydshieh commited on

update model script
b00cdfe

ydshieh commited on

Clean Flax ViT + GPT2-LM script
3ed2a5d

ydshieh commited on

add codes
a244e91

ydshieh commited on