LongK171 commited on
Commit
9ea5b58
·
1 Parent(s): 6577f36

update requirement

Browse files
Files changed (1) hide show
  1. requirements.txt +20 -3
requirements.txt CHANGED
@@ -4,7 +4,6 @@
4
  torch==2.3.1
5
  torchvision==0.18.1
6
  torchaudio==2.3.1
7
- pytorch-cuda==12.1 # nếu dùng conda, pip sẽ bỏ qua
8
 
9
  # ===============================
10
  # Transformers Ecosystem
@@ -25,9 +24,27 @@ rouge-score>=0.1.2
25
  # ===============================
26
  # Semantic Search / FAISS
27
  # ===============================
28
- faiss-gpu==1.8.0
 
 
29
 
30
  # ===============================
31
  # PDF / Text Processing
32
  # ===============================
33
- PyMuPDF==1
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
4
  torch==2.3.1
5
  torchvision==0.18.1
6
  torchaudio==2.3.1
 
7
 
8
  # ===============================
9
  # Transformers Ecosystem
 
24
  # ===============================
25
  # Semantic Search / FAISS
26
  # ===============================
27
+ # Nếu GPU runtime (T4 / A10G) => dùng faiss-gpu
28
+ # Nếu CPU-only => đổi sang faiss-cpu
29
+ faiss-cpu==1.8.0
30
 
31
  # ===============================
32
  # PDF / Text Processing
33
  # ===============================
34
+ PyMuPDF>=1.23.0
35
+ underthesea>=6.8.0
36
+ numpy>=1.26.0
37
+ pandas>=2.2.0
38
+
39
+ # ===============================
40
+ # Web / API
41
+ # ===============================
42
+ fastapi>=0.111.0
43
+ uvicorn[standard]>=0.30.0
44
+
45
+ # ===============================
46
+ # Utils
47
+ # ===============================
48
+ regex>=2024.4.16
49
+ tqdm>=4.66.3
50
+ requests>=2.31.0