Update requirements.txt

requirements.txt

@@ -1,6 +1,9 @@
-torchvision
+--extra-index-url https://download.pytorch.org/whl/cu121
+
+torch==2.4.0
+torchvision==0.19.0
 diffusers
-transformers==4.49.0
+transformers==4.46.3
 einops
 huggingface_hub
 opencv-python
@@ -17,7 +20,9 @@ open3d
 timm
 kornia
 ninja
-https://huggingface.co/spaces/JeffreyXiang/TRELLIS/resolve/main/wheels/nvdiffrast-0.3.3-cp310-cp310-linux_x86_64.whl?download=true
 cvcuda_cu12
 gltflib
-torch-cluster
+torch-cluster
+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/cavargas10/TRELLIS-Imagen3D/resolve/main/wheels/diff_gaussian_rasterization-0.0.0-cp310-cp310-linux_x86_64.whl?download=true
+https://huggingface.co/spaces/cavargas10/TRELLIS-Imagen3D/resolve/main/wheels/nvdiffrast-0.3.3-cp310-cp310-linux_x86_64.whl?download=true