perler commited on
Commit
1365ce4
1 Parent(s): d1c8842
Files changed (1) hide show
  1. requirements.txt +1 -2
requirements.txt CHANGED
@@ -1,6 +1,6 @@
1
  pytorch-lightning>=2.1
2
  # torch_geometric>=2
3
- pyg_lib
4
  torch-cluster>=1.6.0
5
  scikit-learn-intelex>=2024.1.0
6
  numpy>=1.26.4
@@ -23,6 +23,5 @@ jsonargparse[signatures]>=4.27.5
23
 
24
  spaces>=0.23
25
  gradio>=4.19
26
- -f https://data.pyg.org/whl/torch-2.1.0+cu124.html
27
  --index-url https://download.pytorch.org/whl/cu124
28
  --extra-index-url=https://pypi.org/simple
 
1
  pytorch-lightning>=2.1
2
  # torch_geometric>=2
3
+ pyg_lib -f https://data.pyg.org/whl/torch-2.1.0+cu124.html
4
  torch-cluster>=1.6.0
5
  scikit-learn-intelex>=2024.1.0
6
  numpy>=1.26.4
 
23
 
24
  spaces>=0.23
25
  gradio>=4.19
 
26
  --index-url https://download.pytorch.org/whl/cu124
27
  --extra-index-url=https://pypi.org/simple