transformers

865e28f9 - 🚨 Safe-ize pi0 graphs (#50160)

🚨 Safe-ize pi0 graphs (#50160) * safeize pi0 graphs * fix onnx import, slow test for cuda graph * address review: processor stats as arguments, real-checkpoint tests, device-safe gather * move to setup