PR #18356 [FX] Symbolic trace for Bloom

Bloom model can now be traced

michaelbenayoun committed 3 years ago

Bloom traced model can be torch scripted and serialized

michaelbenayoun committed 3 years ago

Bloom can be traced with variable keyword arguments

michaelbenayoun committed 3 years ago

Enable XLNet support

michaelbenayoun committed 3 years ago

Disable XLNet for now

michaelbenayoun committed 3 years ago

transformers [FX] Symbolic trace for Bloom #18356 Merged