alihan/light-the-torch
1
0
Fork 0
mirror of https://github.com/pmeier/light-the-torch.git synced 2024-09-08 23:29:28 +03:00
Code Issues Packages Projects Releases Wiki Activity

exclude torchrec from patched dists (#85)

Browse Source
This commit is contained in:
Philip Meier
2022-07-16 23:09:25 +02:00
committed by GitHub
parent c8d9b1d28d
commit fea88000cb
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
scripts/check_available_pytorch_dists.py
View File

@@ -14,10 +14,13 @@ EXCLUDED_PYTORCH_DIST = {
"torch_cuda80",
"torch_nightly",
"torchaudio_nightly",
"torchrec",
"torchrec_cpu",
"torchrec_nightly",
"torchrec_nightly_3.7_cu11.whl",
"torchrec_nightly_3.8_cu11.whl",
"torchrec_nightly_3.9_cu11.whl",
"torchrec_nightly_cpu",
}
PATCHED_PYTORCH_DISTS = set(PYTORCH_DISTRIBUTIONS)