transformers

cc7803c0 - Register ModelOutput as supported torch pytree nodes (#26618)

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