PR #50160 🚨 Safe-ize pi0 graphs

🚨 Safe-ize pi0 graphs #50160

molbap merged 4 commits into main from pi0-graph-safe-sampling

safeize pi0 graphs

8b2cf909

molbap requested a review from

zucchini-nlp 7 days ago

ArthurZucker approved these changes on 2026-11-02

fix onnx import, slow test for cuda graph

4ffa4db4

zucchini-nlp commented on 2026-11-02

address review: processor stats as arguments, real-checkpoint tests, …

c71dedb6

zucchini-nlp approved these changes on 2026-11-03

molbap changed the title ~~Safe-ize pi0 graphs~~ 🚨 Safe-ize pi0 graphs 6 days ago

molbap commented on 2026-11-03

move to setup

3bcbaa60

molbap merged 865e28f9 into main 6 days ago

molbap deleted the pi0-graph-safe-sampling branch 6 days ago

Reviewers

zucchini-nlp

ArthurZucker

Assignees

No one assigned

Labels

None yet

Milestone

No milestone