Refactor the pytorch_doc_push_script to take a branch
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/23556
Test Plan:
- Run ci
Imported from OSS
Differential Revision: D16563747
Pulled By: zou3519
fbshipit-source-id: 104371b3712c00b073a82e5145090e7bd6fd2d53
