Simon pick Claude commited on
Commit
df1dc9e
·
1 Parent(s): ef3d843

Fix torch-cluster compatibility by pinning PyTorch to stable version

Browse files

Resolves OSError: undefined symbol with torch-cluster by using PyTorch 2.4.0 and matching PyG binaries. This ensures torch-cluster is compiled for the correct PyTorch version.

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude <noreply@anthropic.com>

Files changed (1) hide show
  1. requirements.txt +4 -3
requirements.txt CHANGED
@@ -1,7 +1,8 @@
1
  gradio==4.44.1
2
  gradio_image_prompter
3
  gradio_litmodel3d
4
- torchvision
 
5
  diffusers
6
  transformers==4.49.0
7
  einops
@@ -24,5 +25,5 @@ https://huggingface.co/spaces/JeffreyXiang/TRELLIS/resolve/main/wheels/nvdiffras
24
  cvcuda_cu12
25
  gltflib
26
  https://huggingface.co/spaces/VAST-AI/TripoSG/resolve/main/diso-0.1.4-cp310-cp310-linux_x86_64.whl?download=true
27
- --find-links https://data.pyg.org/whl/torch-2.6.0+cu124.html
28
- torch-cluster
 
1
  gradio==4.44.1
2
  gradio_image_prompter
3
  gradio_litmodel3d
4
+ torch==2.4.0
5
+ torchvision==0.19.0
6
  diffusers
7
  transformers==4.49.0
8
  einops
 
25
  cvcuda_cu12
26
  gltflib
27
  https://huggingface.co/spaces/VAST-AI/TripoSG/resolve/main/diso-0.1.4-cp310-cp310-linux_x86_64.whl?download=true
28
+ --find-links https://data.pyg.org/whl/torch-2.4.0+cu121.html
29
+ torch-cluster