Opt in flax and tf #17388
ArthurZucker
merged
41 commits into
huggingface:main
from
ArthurZucker:opt-flax-tf
initial commit
3bd7a700
add init file
b4e03b7e
update globakl init
5f4ad0ab
update index and dummy objects
9bf9b95e
style
7e534845
update modelling auto
b3a74bde
fix initi typo in src/transformers
a8e76bca
fix typo in modeling tf auto, opt was in wrong mapping name
96417090
fixed a slow test : saved_model
18bc1915
style
12577686
fix positionnal embedding if no position id is provided
6355a10c
update tf test
89d4dc96
update test flax requirements
da57713f
fixed serialization
4fea708d
update
d6b4fd5e
patrickvonplaten
commented
on 2022-05-24
patrickvonplaten
commented
on 2022-05-24
patrickvonplaten
commented
on 2022-05-24
patrickvonplaten
commented
on 2022-05-24
update tf name to allow smooth convertion
2215d1cf
update flax tests
fd180141
style
156b9832
fix test typo
9597454a
fix tf typo test
72510ea9
add xla for generate support in causal LM
353f62f9
fixed bug
5b2bcec8
cleaned tf tests
d8c3c24d
style
b7aa9d25
removed from PT for slow tests
1894095a
fix typp
9431c2cb
opt test as slow
55772ad9
trying to fix GPT2 undefined
159b606c
patrickvonplaten
approved these changes
on 2022-05-25
correct documentation and add to test doc
79f005cd
update tf doc
875a75cd
fix doc
a1bf76c4
fake commit
1a0389bc
patrickvonplaten
commented
on 2022-05-26
Apply suggestions from code review
0ba0740b
update test based on review
93524069
Merge branch 'opt-flax-tf' of https://github.com/ArthurZucker/transfo…
3ba2bbe1
Merge branch 'main' of https://github.com/huggingface/transformers in…
c9ef765c
merged main layer for functionning test
8b3cbd30
fixup + quality
d0549067
gante
approved these changes
on 2022-05-30
Apply suggestions from code review
5f1bb052
update long comment
f9a22020
make fix copies
a6f9537e
Reviewers
Assignees
Labels
Core: Modeling
Milestone
No milestone
Login to write a write a comment.
Login via GitHub