DevPatel0611 commited on
Commit
e90d252
·
1 Parent(s): 6b66148

Use PyTorch CPU index to fix build timeout and memory crash

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -0
requirements.txt CHANGED
@@ -1,3 +1,4 @@
 
1
  pandas>=2.0.0
2
  numpy>=1.24.0
3
  scikit-learn>=1.3.0
 
1
+ --extra-index-url https://download.pytorch.org/whl/cpu
2
  pandas>=2.0.0
3
  numpy>=1.24.0
4
  scikit-learn>=1.3.0