Update VMA. (#47727)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/47727
Test Plan: Imported from OSS
Reviewed By: IvanKobzarev
Differential Revision: D25032739
Pulled By: AshkanAliabadi
fbshipit-source-id: 223df5f18dbfee02ed41eb5e116cc15437e28e8e
