|
44 | 44 |
|
45 | 45 | echo "Using pip executable: $PIP_EXECUTABLE"
|
46 | 46 |
|
47 |
| -# |
48 |
| -# First install requirements in install/requirements.txt. Older torch may be |
49 |
| -# installed from the dependency of other models. It will be overridden by |
50 |
| -# newer version of torch nightly installed later in this script. |
51 |
| -# |
52 |
| - |
53 |
| -( |
54 |
| - set -x |
55 |
| - $PIP_EXECUTABLE install -r install/requirements.txt --extra-index-url https://download.pytorch.org/whl/nightly/cu121 |
56 |
| -) |
57 |
| - |
58 | 47 | # Since torchchat often uses main-branch features of pytorch, only the nightly
|
59 | 48 | # pip versions will have the required features. The PYTORCH_NIGHTLY_VERSION value should
|
60 | 49 | # agree with the third-party/pytorch pinned submodule commit.
|
61 | 50 | #
|
62 | 51 | # NOTE: If a newly-fetched version of the executorch repo changes the value of
|
63 | 52 | # PYTORCH_NIGHTLY_VERSION, you should re-run this script to install the necessary
|
64 | 53 | # package versions.
|
65 |
| -PYTORCH_NIGHTLY_VERSION=dev20241213 |
| 54 | +PYTORCH_NIGHTLY_VERSION=dev20241218 |
66 | 55 |
|
67 | 56 | # Nightly version for torchvision
|
68 |
| -VISION_NIGHTLY_VERSION=dev20241213 |
| 57 | +VISION_NIGHTLY_VERSION=dev20241218 |
69 | 58 |
|
70 | 59 | # Nightly version for torchtune
|
71 |
| -TUNE_NIGHTLY_VERSION=dev20241126 |
| 60 | +TUNE_NIGHTLY_VERSION=dev20241218 |
72 | 61 |
|
73 | 62 | # Uninstall triton, as nightly will depend on pytorch-triton, which is one and the same
|
74 | 63 | (
|
@@ -96,6 +85,16 @@ REQUIREMENTS_TO_INSTALL=(
|
96 | 85 | torchtune=="0.5.0.${TUNE_NIGHTLY_VERSION}"
|
97 | 86 | )
|
98 | 87 |
|
| 88 | +# |
| 89 | +# First install requirements in install/requirements.txt. Older torch may be |
| 90 | +# installed from the dependency of other models. It will be overridden by |
| 91 | +# newer version of torch nightly installed later in this script. |
| 92 | +# |
| 93 | +( |
| 94 | + set -x |
| 95 | + $PIP_EXECUTABLE install -r install/requirements.txt --extra-index-url "${TORCH_NIGHTLY_URL}" |
| 96 | +) |
| 97 | + |
99 | 98 | # Install the requirements. --extra-index-url tells pip to look for package
|
100 | 99 | # versions on the provided URL if they aren't available on the default URL.
|
101 | 100 | (
|
|
0 commit comments