-
Bowen Bao authored
* Register BaseOutput subclasses as supported torch.utils._pytree nodes * lint --------- Co-authored-by:Dhruv Nair <dhruv.nair@gmail.com>
c7617e48
* Register BaseOutput subclasses as supported torch.utils._pytree nodes
* lint
---------
Co-authored-by: Dhruv Nair <dhruv.nair@gmail.com>