SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity
Files
68ae3be7f514b0e0411664e2a2cdbc6264fd3a0b
HuggingFace_transformer/tests/utils/test_model_output.py
Xuehai Pan cc7803c0a6 Register ModelOutput as supported torch pytree nodes (#26618) 
* Register ModelOutput as supported torch pytree nodes

* Test ModelOutput as supported torch pytree nodes

* Update type hints for pytree unflatten functions
2023-10-24 11:02:40 +02:00

5.6 KiB
Raw Blame History

View Raw
Reference in New Issue View Git Blame Copy Permalink