Update requirements.txt to specify versions for torch, torchvision, xformers, spconv-cu120, transformers, and gradio_litmodel3d

requirements.txt

@@ -1,8 +1,11 @@
+--extra-index-url https://download.pytorch.org/whl/cu121
+
 accelerate
 sentencepiece
 diffusers
-torch
-torchvision
+
+torch==2.4.0
+torchvision==0.19.0
 pillow==10.4.0
 imageio==2.36.1
 imageio-ffmpeg==0.5.1
@@ -17,10 +20,11 @@ xatlas==0.0.9
 pyvista==0.44.2
 pymeshfix==0.17.0
 igraph==0.11.8
-xformers
-spconv-cu120
-transformers
-gradio_litmodel3d
 git+https://github.com/EasternJournalist/utils3d.git@9a4eb15e4021b67b12c460c7057d642626897ec8
+xformers==0.0.27.post2
+spconv-cu120==2.3.6
+transformers==4.46.3
+gradio_litmodel3d==0.0.1
+https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.0.post2/flash_attn-2.7.0.post2+cu12torch2.4cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
 https://huggingface.co/spaces/JeffreyXiang/TRELLIS/resolve/main/wheels/diff_gaussian_rasterization-0.0.0-cp310-cp310-linux_x86_64.whl?download=true
 https://huggingface.co/spaces/JeffreyXiang/TRELLIS/resolve/main/wheels/nvdiffrast-0.3.3-cp310-cp310-linux_x86_64.whl?download=true