Diksha2001 commited on
Commit
546c728
·
verified ·
1 Parent(s): 09ff480

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +18 -9
requirements.txt CHANGED
@@ -1,26 +1,35 @@
1
- azure-storage-blob==12.14.1
2
- runpod
3
- torch==2.3.1
4
- xformers==0.0.27
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
- packaging==23.0
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
+