Register BaseOutput subclasses as supported torch.utils._pytree nodes #5459
Register BaseOutput subclasses as supported torch.utils._pytree nodes
368de363
Merge branch 'main' into bowbao/pytree_base_output
6646348a
DN6
requested a review
from
DN6
2 years ago
DN6
approved these changes
on 2023-10-20
lint
6f76eb4c
DN6
merged
c7617e48
into main 2 years ago
Login to write a write a comment.
Login via GitHub