uyen1109 commited on
Commit
6f0d11d
·
verified ·
1 Parent(s): afeff59

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +9 -2
requirements.txt CHANGED
@@ -1,7 +1,14 @@
 
 
 
 
 
 
 
 
 
1
  gradio>=4.44.0
2
  huggingface_hub>=0.24.6
3
- torch==2.3.1
4
  numpy<2
5
  h5py
6
  matplotlib
7
- # Không dùng PyG trên Space để tránh lỗi build.
 
1
+ torch==2.3.1
2
+ --find-links https://data.pyg.org/whl/torch-2.3.1+cpu.html
3
+ torch-scatter==2.1.2
4
+ torch-sparse==0.6.18
5
+ torch-cluster==1.6.3
6
+ torch-spline-conv==1.2.2
7
+ pyg-lib==0.4.0
8
+ torch-geometric==2.5.3
9
+
10
  gradio>=4.44.0
11
  huggingface_hub>=0.24.6
 
12
  numpy<2
13
  h5py
14
  matplotlib