stream pkl (#79931)
Summary: Let the stream version of pickle loader to short circuit early
Test Plan:
this might affects loading peak mem
before:
Differential Revision: D37304855
Pull Request resolved: https://github.com/pytorch/pytorch/pull/79931
Approved by: https://github.com/junesg, https://github.com/davidberard98