jkorstad commited on
Commit
cd5a646
·
verified ·
1 Parent(s): c01c482

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +10 -43
requirements.txt CHANGED
@@ -1,47 +1,14 @@
1
- # PyTorch (UniRig: >=2.3.1)
2
- torch>=2.3.1
3
- torchvision>=0.18.1 # Compatible with torch 2.3.1
4
-
5
- # PyTorch Geometric libraries (for CUDA 12.1, matching PyTorch 2.3.1)
6
- torch-scatter -f https://data.pyg.org/whl/torch-2.3.1+cu121.html
7
- torch-cluster -f https://data.pyg.org/whl/torch-2.3.1+cu121.html
8
-
9
- # Spconv (for CUDA 12.1)
10
- spconv-cu121
11
 
12
- # Numpy (UniRig: 1.26.4)
13
- numpy==1.26.4
14
-
15
- # Gradio (Updated to 5.29.0)
16
- gradio>=5.29.0,<5.30.0
17
 
 
 
 
18
  spaces
19
 
20
- # Dependencies from UniRig's official requirements.txt
21
- transformers
22
- python-box
23
- einops
24
- omegaconf
25
- pytorch_lightning
26
- lightning
27
- addict
28
- timm
29
- fast-simplification
30
- # Replaced flash_attn with pre-built wheel
31
- # https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.4/flash_attn-2.7.4+cu121torch2.3.1-cp310-cp310-linux_x86_64.whl
32
- trimesh
33
- open3d
34
- pyrender
35
- huggingface_hub
36
- scipy
37
- smplx
38
- chumpy
39
- opencv-python
40
- Pillow
41
- pyquaternion
42
- hydra-core
43
- rich
44
- distinctipy
45
- scikit-image
46
- mediapipe
47
- bpy
 
1
+ # Main Gradio/Space dependencies (Python 3.10 for ZeroGPU compatibility)
 
 
 
 
 
 
 
 
 
2
 
3
+ # PyTorch for the main environment (ZeroGPU often requires torch)
4
+ # Ensure this version is compatible with Gradio/Spaces and Python 3.10
5
+ torch>=2.3.1
6
+ torchvision>=0.18.1
 
7
 
8
+ # Gradio and Hugging Face Spaces helper
9
+ # Lock Gradio version for stability if needed, e.g., ==5.29.0
10
+ gradio>=5.29.0,<5.30.0
11
  spaces
12
 
13
+ # NOTE: All UniRig-specific dependencies have been moved to unirig_requirements.txt
14
+ # NOTE: 'bpy' should NOT be listed here; it's part of the Blender environment.