Fix jit doc model loading example (#48104)
Summary:
Fixes #{issue number}
Pull Request resolved: https://github.com/pytorch/pytorch/pull/48104
Reviewed By: jamesr66a
Differential Revision: D25028353
Pulled By: suo
fbshipit-source-id: aaf74a40e7150a278d100e129740cfe1cef99af2