Mock model loading in `test_run_locally`

[viswavi]

In our integration test, we're currently loading a model in the Trainer. this should be mocked to avoid having to download a full transformers model each time this test is run on a new machine.