transformers
4e2f4a92 - [FX] Symbolic trace for Bloom (#18356)

Commit
3 years ago
[FX] Symbolic trace for Bloom (#18356) * Bloom model can now be traced * Bloom traced model can be torch scripted and serialized * Bloom can be traced with variable keyword arguments * Enable XLNet support * Disable XLNet for now
Parents
Loading