Spaces:
Paused
Paused
Diksha2001
commited on
Update requirements.txt
Browse files- requirements.txt +18 -9
requirements.txt
CHANGED
@@ -1,26 +1,35 @@
|
|
1 |
-
|
2 |
-
|
3 |
-
|
4 |
-
|
5 |
transformers>=4.35.0
|
6 |
accelerate>=0.34.1
|
7 |
bitsandbytes>=0.43.3
|
8 |
trl>=0.7.1
|
9 |
peft>=0.7.0
|
|
|
10 |
triton
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
11 |
PyPDF2==2.11.2
|
12 |
PyMuPDF
|
13 |
pytesseract==0.3.10
|
14 |
Pillow>=9.0.0
|
|
|
|
|
15 |
openai>=1.0.0
|
16 |
datasets>=2.13.1
|
17 |
-
|
18 |
-
tqdm==4.66.1
|
19 |
-
git+https://github.com/unslothai/unsloth.git#egg=unsloth[colab-new]
|
20 |
sentence-transformers
|
21 |
-
vllm==0.6.4.post1
|
22 |
rouge-score==0.1.2
|
23 |
-
huggingface-hub==0.26.1
|
24 |
gradio
|
25 |
nltk
|
26 |
requests
|
|
|
|
1 |
+
# Core ML/AI Libraries
|
2 |
+
torch
|
3 |
+
torchvision
|
4 |
+
torchaudio
|
5 |
transformers>=4.35.0
|
6 |
accelerate>=0.34.1
|
7 |
bitsandbytes>=0.43.3
|
8 |
trl>=0.7.1
|
9 |
peft>=0.7.0
|
10 |
+
xformers
|
11 |
triton
|
12 |
+
unsloth @ git+https://github.com/unslothai/unsloth.git#egg=unsloth[colab-new]
|
13 |
+
|
14 |
+
# Azure Storage
|
15 |
+
azure-storage-blob==12.14.1
|
16 |
+
|
17 |
+
# Deployment & Serverless
|
18 |
+
runpod
|
19 |
+
|
20 |
+
# PDF Processing
|
21 |
PyPDF2==2.11.2
|
22 |
PyMuPDF
|
23 |
pytesseract==0.3.10
|
24 |
Pillow>=9.0.0
|
25 |
+
|
26 |
+
# AI & ML Utilities
|
27 |
openai>=1.0.0
|
28 |
datasets>=2.13.1
|
29 |
+
vllm==0.6.2 # Adjusted for compatibility
|
|
|
|
|
30 |
sentence-transformers
|
|
|
31 |
rouge-score==0.1.2
|
|
|
32 |
gradio
|
33 |
nltk
|
34 |
requests
|
35 |
+
|