manoskary commited on
Commit
9de1f53
·
verified ·
1 Parent(s): 5e791e7

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +29 -22
requirements.txt CHANGED
@@ -1,22 +1,32 @@
1
- # PyTorch (install first, required by other packages)
2
- # Use CPU-only version for compatibility; change index URL if you need CUDA
3
- --index-url https://download.pytorch.org/whl/cpu
4
- torch>=2.0.0
5
- torchvision>=0.15.0
6
-
7
- # PyTorch Geometric extensions (installed without build isolation to use existing torch)
8
- # Note: These may require compilation; use --no-build-isolation flag when installing
9
- torch-scatter>=2.1.0
10
- torch-sparse>=0.6.0
11
- torch-cluster>=1.6.0
12
- torch-spline-conv>=1.2.0
13
-
14
- # Reset to default PyPI index for remaining packages
 
 
 
 
 
 
 
 
 
 
15
  --index-url https://pypi.org/simple
16
 
17
- # PyTorch extensions
18
  pytorch-lightning>=2.0.0
19
- torch-geometric>=2.3.0
20
 
21
  # Data processing
22
  numpy>=1.21.0
@@ -26,10 +36,10 @@ pandas>=1.5.0
26
  partitura>=1.4.0
27
  music21>=8.0.0
28
 
29
- # Gradio and Spaces
30
  gradio>=4.0.0
31
 
32
- # Wandb for model loading
33
  wandb>=0.13.0
34
 
35
  # PDF rendering and image processing
@@ -39,10 +49,7 @@ Pillow>=9.0.0
39
 
40
  # ML utilities
41
  scikit-learn>=1.1.0
42
- torchmetrics>=0.11.0
43
-
44
- # Other utilities
45
  tqdm>=4.64.0
46
 
47
  # Graph music representation library (required by AnalysisGNN)
48
- git+https://github.com/manoskary/graphmuse.git
 
1
+ ##########################################################
2
+ # TORCH + TORCH GEOMETRIC (CPU BUILD)
3
+ ##########################################################
4
+
5
+ # Install torch (CPU version)
6
+ -f https://download.pytorch.org/whl/cpu
7
+ torch==2.3.1+cpu
8
+ torchvision==0.18.1+cpu
9
+ torchaudio==2.3.1+cpu
10
+
11
+ # Install PyTorch Geometric dependencies (CPU wheels)
12
+ -f https://data.pyg.org/whl/torch-2.3.1+cpu.html
13
+ torch-scatter==2.1.2
14
+ torch-sparse==0.6.18
15
+ torch-cluster==1.6.3
16
+ torch-spline-conv==1.2.2
17
+
18
+ # Main PyTorch Geometric library
19
+ torch-geometric==2.5.3
20
+
21
+ ##########################################################
22
+ # RESET TO DEFAULT PYPI INDEX FOR REMAINING PACKAGES
23
+ ##########################################################
24
+
25
  --index-url https://pypi.org/simple
26
 
27
+ # PyTorch ecosystem
28
  pytorch-lightning>=2.0.0
29
+ torchmetrics>=0.11.0
30
 
31
  # Data processing
32
  numpy>=1.21.0
 
36
  partitura>=1.4.0
37
  music21>=8.0.0
38
 
39
+ # Gradio and Hugging Face Spaces
40
  gradio>=4.0.0
41
 
42
+ # Experiment tracking
43
  wandb>=0.13.0
44
 
45
  # PDF rendering and image processing
 
49
 
50
  # ML utilities
51
  scikit-learn>=1.1.0
 
 
 
52
  tqdm>=4.64.0
53
 
54
  # Graph music representation library (required by AnalysisGNN)
55
+ git+https://github.com/manoskary/graphmuse.git