Fix delegating to jit.load from torch.load (#40937)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/40937
Test Plan: Imported from OSS
Differential Revision: D22363816
Pulled By: jamesr66a
fbshipit-source-id: 50fc318869407fe8b215368026eaceb129b68a46
