deepsu commited on
Commit
cc3f682
·
1 Parent(s): e86b059

CPU only use

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -1
requirements.txt CHANGED
@@ -6,7 +6,8 @@ pandas
6
  scikit-learn
7
  pypdf
8
 
9
- torch
 
10
  transformers
11
  sentence-transformers
12
  peft
 
6
  scikit-learn
7
  pypdf
8
 
9
+ torch --index-url https://download.pytorch.org/whl/cpu
10
+
11
  transformers
12
  sentence-transformers
13
  peft