transformers
f4a0d6ff - A cleaner and more scalable implementation of symbolic tracing (#11763)

Commit
4 years ago
A cleaner and more scalable implementation of symbolic tracing (#11763) Cleaner and more scalable implementation of symbolic tracing with torch.fx, and provides support for new architectures: - ALBERT - DistilBERT - MobileBERT - MegatronBERT - GPT2 - GPT Neo Co-authored-by: Michael Benayoun <michael@huggingface.co>
Parents
Loading