Followup fix after #78828 (#79554)
Will be skipped when imported internally, for more details see https://www.internalfb.com/diff/D37114156?src_version_fbid=3331368873807344
Pull Request resolved: https://github.com/pytorch/pytorch/pull/79554
Approved by: https://github.com/albanD
