tests/models/bloom/test_modeling_bloom.py · 153d1361c7dcc91c7735cae73e1f594cfcab3e21 · chenpangpang / transformers

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

Michael Benayoun authored Jul 29, 2022

* 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

4e2f4a92

test_modeling_bloom.py 32.8 KB

Replace test_modeling_bloom.py