abdull4h commited on
Commit
aabcc85
·
verified ·
1 Parent(s): ff97ad1

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +29 -7
requirements.txt CHANGED
@@ -1,8 +1,30 @@
 
1
  gradio==4.44.0
2
- transformers==4.36.0
3
- torch==2.1.0
4
- accelerate==0.25.0
5
- sentencepiece==0.1.99
6
- protobuf==4.25.1
7
- numpy==1.24.3
8
- pandas==2.0.3
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Core Gradio and ML dependencies
2
  gradio==4.44.0
3
+ transformers==4.40.0
4
+ torch==2.2.0
5
+ accelerate==0.30.0
6
+
7
+ # Text processing and tokenization
8
+ sentencepiece==0.2.0
9
+ tokenizers==0.19.1
10
+
11
+ # Data handling
12
+ numpy==1.26.4
13
+ pandas==2.2.0
14
+
15
+ # Protocol buffers (for model serialization)
16
+ protobuf==4.25.3
17
+
18
+ # GPU optimization and CUDA support
19
+ nvidia-ml-py3==12.535.161
20
+
21
+ # Additional utilities for better performance
22
+ psutil==5.9.8
23
+ packaging==24.0
24
+
25
+ # Optional: For better model management
26
+ huggingface-hub==0.20.3
27
+ safetensors==0.4.2
28
+
29
+ # For error handling and logging
30
+ tqdm==4.66.2