update

requirements.txt

@@ -1,12 +1,17 @@
---extra-index-url https://download.pytorch.org/whl/cu121
+--extra-index-url https://download.pytorch.org/whl/cu126
 
-torch==2.4.0
-torchvision==0.19.0
-xformers==0.0.27.post2
+setuptools==80.10.2
+wheel
+packaging
+Cython>=0.29.37
+
+torch==2.8.0
+torchvision==0.23.0
+xformers==0.0.32.post2
 pytorch-lightning==2.4.0
-spconv-cu120==2.3.6
+spconv-cu126==2.3.7
 numpy==1.26.4
-triton
+triton==3.4.0
 dataclasses_json
 easydict
 opencv-python>4.5
@@ -21,7 +26,7 @@ pyvista==0.36.1
 openai==1.58.1
 transformers==4.42.4
 gradio==5.48.0
-spaces==0.47.0
+spaces
 sentencepiece==0.2.0
 diffusers==0.31.0
 xatlas==0.0.9
@@ -59,15 +64,8 @@ timm
 open3d
 MoGe@git+https://github.com/microsoft/MoGe.git@a8c3734
 
-
-# git+https://github.com/facebookresearch/pytorch3d.git@stable
-https://huggingface.co/xinjjj/RoboAssetGen/resolve/main/wheel_cu121/pytorch3d-0.7.8-cp310-cp310-linux_x86_64.whl
-# git+https://github.com/nerfstudio-project/gsplat.git@v1.5.3
-https://github.com/nerfstudio-project/gsplat/releases/download/v1.5.0/gsplat-1.5.0+pt24cu121-cp310-cp310-linux_x86_64.whl
-# flash-attn==2.7.0.post2
-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
-# kaolin@git+https://github.com/NVIDIAGameWorks/kaolin.git@v0.16.0
-https://huggingface.co/xinjjj/RoboAssetGen/resolve/main/wheel_cu121/kaolin-0.16.0-cp310-cp310-linux_x86_64.whl
-# nvdiffrast@git+https://github.com/NVlabs/nvdiffrast.git#egg=729261d
-https://huggingface.co/xinjjj/RoboAssetGen/resolve/main/wheel_cu121/nvdiffrast-0.3.3-cp310-cp310-linux_x86_64.whl
-https://huggingface.co/xinjjj/RoboAssetGen/resolve/main/wheel_cu121/diff_gaussian_rasterization-0.0.0-cp310-cp310-linux_x86_64.whl
+https://huggingface.co/xinjjj/RoboAssetGen/resolve/main/wheel_cu126/diff_gaussian_rasterization-0.0.0-cp310-cp310-linux_x86_64.whl
+https://huggingface.co/xinjjj/RoboAssetGen/resolve/main/wheel_cu126/gsplat-1.5.3-cp310-cp310-linux_x86_64.whl
+https://huggingface.co/xinjjj/RoboAssetGen/resolve/main/wheel_cu126/kaolin-0.18.0-cp310-cp310-linux_x86_64.whl
+https://huggingface.co/xinjjj/RoboAssetGen/resolve/main/wheel_cu126/nvdiffrast-0.3.3-py3-none-any.whl
+https://huggingface.co/xinjjj/RoboAssetGen/resolve/main/wheel_cu126/pytorch3d-0.7.8-cp310-cp310-linux_x86_64.whl