tests/test_modeling_gpt_neo.py · f4a0d6ff867e8a82a33d7a653e7d45372a463271 · chenpangpang / transformers

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

Michael Benayoun authored May 20, 2021



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>

f4a0d6ff

test_modeling_gpt_neo.py 27.5 KB

Replace test_modeling_gpt_neo.py