Skip to content

Commit

Permalink
Revert "[Temp] Use Meta CDN for building domain binaries" (#5762)
Browse files Browse the repository at this point in the history
Reverts #5747

Reverting due to: pytorch/ao#1063
  • Loading branch information
atalman authored Oct 14, 2024
1 parent f08226e commit 38d9358
Showing 1 changed file with 1 addition and 4 deletions.
5 changes: 1 addition & 4 deletions tools/pkg-helpers/pytorch_pkg_helpers/wheel.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,15 +18,12 @@ def get_pytorch_pip_install_command(
channel: str,
) -> List[str]:
torch_pkg = "torch"
download_pytorch_org = "download.pytorch.org"
if pytorch_version != "":
torch_pkg += f"=={pytorch_version}"
pip_install = f"pip install {torch_pkg}"
if channel == "nightly":
pip_install += " --pre"
# Use Meta CDN for nightly builds
download_pytorch_org = "d3kup0pazkvub8.cloudfront.net"
extra_index = f"https://{download_pytorch_org}/whl/{channel}/{gpu_arch_version}"
extra_index = f"https://download.pytorch.org/whl/{channel}/{gpu_arch_version}"
return [f"export PIP_INSTALL_TORCH='{pip_install} --index-url {extra_index}'"]


Expand Down

0 comments on commit 38d9358

Please sign in to comment.