save tokenizer to each ckpt folder
6d4331f
ydshieh
commited on
Commit History
Add usage example
43c04d5
ydshieh
commited on
Save prediction results
820b2a2
ydshieh
commited on
Add a script to create dummy pretrained models for testing
fd8c682
ydshieh
commited on
Add dataset usage example script
8ffa189
ydshieh
commited on
Add dataset script
9628bd8
ydshieh
commited on
Rename file
4c37b45
ydshieh
commited on
Use new FlaxVisionEncoderDecoderModel class
3bffad7
ydshieh
commited on
copy run_summarization_flax.py
a1a9885
ydshieh
commited on
remove dataset script
ae50e80
ydshieh
commited on
Update dataset script
0bf3646
ydshieh
commited on
Fix style
64afcd5
ydshieh
commited on
Update test_model.py
ec3ceb6
ydshieh
commited on
Change a parameter in decode(): deterministic -> train
7d3b1a0
ydshieh
commited on
remove unused scripts
e5b3a97
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
update check against pytorch's version
06bcf58
ydshieh
commited on
Add a test against pytorch's GPT2
3b81fb5
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
update test_model.py
f082d66
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
Merge branch 'main' of https://huggingface.co/flax-community/vit-gpt2 into main
7b292b7
ydshieh
commited on
update test_model.py
845642f
ydshieh
commited on
Merge branch 'main' of https://huggingface.co/flax-community/vit-gpt2
56fa8ac
ydshieh
commited on
Make a remark about a serious bug I made previously
0ac6b6e
ydshieh
commited on
update test_model.py
dc74cb9
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