Temp fix for upgrader (#78589)
Summary: see: https://fb.workplace.com/groups/pytorch.edge.users/permalink/1086950342175258/
Test Plan: https://www.internalfb.com/phabricator/paste/view/P503629793
Reviewed By: qihqi
Differential Revision: D36694318
Pull Request resolved: https://github.com/pytorch/pytorch/pull/78589
Approved by: https://github.com/qihqi