Fix CI quick-checks (#71773)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/71773
Test Plan: Imported from OSS
Reviewed By: pbelevich
Differential Revision: D33770042
Pulled By: anjali411
fbshipit-source-id: 9dd3f8c8592663d385ab0cd4376aaa4b9c7d9ec2
(cherry picked from commit 739c8885c78b3e39c0b5814f1bece0e3bbb6521b)