olivercareyncl commited on
Commit
fcfd8b2
·
verified ·
1 Parent(s): d428d7f

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +14 -4
requirements.txt CHANGED
@@ -1,16 +1,26 @@
1
- torch
2
- torchvision
3
- torchaudio
 
 
 
4
  gradio==4.44.1
5
  gradio_litmodel3d
6
  gradio_image_prompter
7
  diffusers
8
  transformers
9
  einops
 
 
 
 
10
  torch-cluster
 
 
 
11
  huggingface_hub
12
  numpy
13
  trimesh
14
  Pillow
15
- skimage
16
  spaces
 
1
+ # Install PyTorch first
2
+ torch==2.1.0
3
+ torchvision==0.16.0
4
+ torchaudio==2.1.0
5
+
6
+ # Core Dependencies
7
  gradio==4.44.1
8
  gradio_litmodel3d
9
  gradio_image_prompter
10
  diffusers
11
  transformers
12
  einops
13
+
14
+ # GPU-based Libraries (Install after torch)
15
+ torch-scatter
16
+ torch-sparse
17
  torch-cluster
18
+ torch-geometric
19
+
20
+ # Other required dependencies
21
  huggingface_hub
22
  numpy
23
  trimesh
24
  Pillow
25
+ scikit-image
26
  spaces