temporary fix for jit test backward compatibility issues
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/31949
Test Plan: Imported from OSS
Differential Revision: D19314763
Pulled By: albanD
fbshipit-source-id: b5eff0ed53a371d260596ca85d914c8bddb0a8aa