update

pre_build_script.sh

@@ -0,0 +1,9 @@
+pip install torch==2.1.0 torchvision==0.16.0 --index-url https://download.pytorch.org/whl/cu118
+pip install xformers==0.0.22.post4 --index-url https://download.pytorch.org/whl/cu118
+pip install torch-scatter -f https://data.pyg.org/whl/torch-2.1.0+cu118.html
+
+pip install -r requirements.txt
+pip install git+https://github.com/NVlabs/nvdiffrast
+pip install git+https://github.com/facebookresearch/pytorch3d.git
+pip install git+https://github.com/Baijiong-Lin/LoRA-Torch
+pip install git+https://github.com/facebookresearch/segment-anything.git

requirements.txt

@@ -1,7 +1,3 @@
-torch==2.1.0
-torchvision==0.16.0
-xformers==0.0.22.post4
-torch-scatter
 omegaconf
 rm_anime_bg
 diffusers==0.27.2
@@ -22,7 +18,3 @@ scikit-learn
 pygltflib
 pymeshlab==2022.2.post3
 pytorch_lightning
-git+https://github.com/NVlabs/nvdiffrast
-git+https://github.com/facebookresearch/pytorch3d.git
-git+https://github.com/Baijiong-Lin/LoRA-Torch
-git+https://github.com/facebookresearch/segment-anything.git