tests/utils/test_model_output.py · cc7803c0a6194ce795ab903979dde9216c82f5bc · chenpangpang / transformers

Xuehai Pan authored Oct 24, 2023

* Register ModelOutput as supported torch pytree nodes

* Test ModelOutput as supported torch pytree nodes

* Update type hints for pytree unflatten functions

cc7803c0

test_model_output.py 5.63 KB

Replace test_model_output.py