change et branch (#608)

* change et branch

* don't use pip for ET

* torch_version

* extra url

* extra index url

* reqs

* load cuda version

* rm torch from req's

---------

Co-authored-by: Michael Gschwind <[email protected]>
Co-authored-by: Michael Gschwind <[email protected]>

Author: 3 people

install_requirements.sh

@@ -30,7 +30,7 @@ fi
 # newer version of torch nightly installed later in this script.
 #
 
-$PIP_EXECUTABLE install -r requirements.txt
+$PIP_EXECUTABLE install -r requirements.txt --extra-index-url https://download.pytorch.org/whl/nightly/cu121
 
 # Since torchchat often uses main-branch features of pytorch, only the nightly
 # pip versions will have the required features. The NIGHTLY_VERSION value should

requirements.txt

@@ -1,9 +1,9 @@
 # Requires python >=3.10
 
 # PyTorch ecosystem
-torch
+#torch==2.4.0.dev20240422
 torchao
-executorch
+# executorch==0.1.2
 
 # Hugging Face download
 huggingface_hub

scripts/install_utils.sh

@@ -11,7 +11,7 @@ install_pip_dependencies() {
   echo "Intalling common pip packages"
   pip3 install wheel "cmake>=3.19" ninja zstd
   pushd ${TORCHCHAT_ROOT}
-  pip3 install -r ./requirements.txt
+  pip3 install -r ./requirements.txt --extra-index-url https://download.pytorch.org/whl/nightly/cu121
   popd
 }
 
@@ -27,7 +27,7 @@ clone_executorch() {
   pushd ${TORCHCHAT_ROOT}/${ET_BUILD_DIR}/src
   git clone https://github.com/pytorch/executorch.git
   cd executorch
-  git checkout viable/strict
+  git checkout main
   echo "Install executorch: submodule update"
   git submodule sync
   git submodule update --init