Meena commited on
Commit
002fb26
1 Parent(s): 6168d97

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +5 -2
requirements.txt CHANGED
@@ -1,6 +1,9 @@
1
- streamlit
2
- torch
 
 
3
  torch-scatter
 
4
  transformers
5
  tensorflow==2.7.0
6
  tensorflow_probability
 
1
+
2
+ -f https://download.pytorch.org/whl/cpu/torch_stable.html
3
+ -f https://data.pyg.org/whl/torch-1.10.0+cu113.html
4
+ torch==1.10.0+cu113
5
  torch-scatter
6
+ streamlit
7
  transformers
8
  tensorflow==2.7.0
9
  tensorflow_probability