Michael Benayoun f4a0d6ff86 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>

2021-05-20 18:02:29 +02:00

15 KiB

Raw Blame History

View Raw