Fix servicelab test failures
Summary:
X-link: https://github.com/pytorch/pytorch/pull/130781
Fix the servicelab test failures of the following models:
- ads_mc_cmf_mc3_v0
- ads_mc_dpa_mc3_v0
- ads_mc_afoc_mc3_v0
- ads_mc_mai_mc3_v0
- ig_feed_over
- ig_mtml_over
Reviewed By: desertfire
Differential Revision: D59704063
fbshipit-source-id: 9264c0af65aabe08704e70f456246714e4b7b70e
