transformers
Register ModelOutput as supported torch pytree nodes
#26618

Merged

Register ModelOutput as supported torch pytree nodes #26618

LysandreJik merged 3 commits into huggingface:main from XuehaiPan:model-output-torch-pytree-node

XuehaiPan force pushed 2 years ago

57d2c218

Test ModelOutput as supported torch pytree nodes

96f94a2b

Update type hints for pytree unflatten functions

44a73ceb

XuehaiPan force pushed to 44a73ceb 2 years ago

ydshieh requested a review from

ydshieh 2 years ago

ydshieh commented on 2023-10-24

ydshieh approved these changes on 2023-10-24

LysandreJik approved these changes on 2023-10-24

LysandreJik merged cc7803c0 into main 2 years ago

XuehaiPan deleted the model-output-torch-pytree-node branch 2 years ago

Reviewers

LysandreJik

ydshieh

Assignees

No one assigned

Labels

None yet

Milestone

No milestone