PR #18356 [FX] Symbolic trace for Bloom

[FX] Symbolic trace for Bloom #18356

michaelbenayoun merged 5 commits into huggingface:main from michaelbenayoun:fx_bloom

Bloom model can now be traced

20c80ba1

Bloom traced model can be torch scripted and serialized

56a387b2

Bloom can be traced with variable keyword arguments

662906ab

Enable XLNet support

02cba6e3

michaelbenayoun requested a review from

sgugger 3 years ago

michaelbenayoun requested a review from

younesbelkada 3 years ago

mfuntowicz commented on 2022-07-29

sgugger approved these changes on 2022-07-29

Disable XLNet for now

a5e10632

younesbelkada commented on 2022-07-29

younesbelkada approved these changes on 2022-07-29

michaelbenayoun merged 4e2f4a92 into main 3 years ago

michaelbenayoun deleted the fx_bloom branch 3 years ago

Reviewers

younesbelkada

sgugger

mfuntowicz

Assignees

No one assigned

Labels

None yet

Milestone

No milestone