fix: debug CUDA issue and update dependencies

app.py

@@ -249,9 +249,16 @@ with gr.Blocks() as demo:
 # Launch the Gradio app
 if __name__ == "__main__":
     pipeline = TrellisImageTo3DPipeline.from_pretrained("JeffreyXiang/TRELLIS-image-large")
-    pipeline.cuda()
+    if torch.cuda.is_available():
+        pipeline.cuda()
+        print("CUDA is available. Using GPU.")
+    else:
+        print("CUDA not available. Falling back to CPU.")
     try:
         pipeline.preprocess_image(Image.fromarray(np.zeros((512, 512, 3), dtype=np.uint8)))    # Preload rembg
     except:
         pass
-    demo.launch()
+    print(f"CUDA Available: {torch.cuda.is_available()}")
+    print(f"CUDA Version: {torch.version.cuda}")
+    print(f"Number of GPUs: {torch.cuda.device_count()}")
+    demo.launch(debug=True)

requirements.txt

@@ -1,6 +1,3 @@
---extra-index-url https://download.pytorch.org/whl/cu121
---find-links https://nvidia-kaolin.s3.us-east-2.amazonaws.com/torch-2.4.0_cu121.html
-
 torch==2.4.0
 torchvision==0.19.0
 pillow==10.4.0
@@ -20,7 +17,6 @@ igraph==0.11.8
 git+https://github.com/EasternJournalist/utils3d.git@9a4eb15e4021b67b12c460c7057d642626897ec8
 xformers==0.0.27.post2
 kaolin==0.17.0
-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