AIdeaText commited on
Commit
cf334f7
·
verified ·
1 Parent(s): be5611a

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +27 -2
requirements.txt CHANGED
@@ -1,2 +1,27 @@
1
- transformers
2
- torch
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Core dependencies
2
+ streamlit>=1.24.0
3
+ torch>=2.0.0
4
+ transformers>=4.31.0
5
+ accelerate>=0.21.0
6
+
7
+ # Optimization and utilities
8
+ bitsandbytes>=0.41.0
9
+ sentencepiece>=0.1.99
10
+ protobuf>=3.20.0
11
+ einops>=0.6.1
12
+
13
+ # Optional but recommended for better performance
14
+ nvidia-cuda-runtime-cu11>=11.7.99
15
+ nvidia-cudnn-cu11>=8.5.0.96
16
+ nvidia-cublas-cu11>=11.10.3.66
17
+
18
+ # Memory optimization
19
+ psutil>=5.9.0
20
+ py3nvml>=0.2.7
21
+
22
+ # Formatting and typing
23
+ black>=23.3.0
24
+ mypy>=1.3.0
25
+
26
+ # UI enhancements
27
+ streamlit-chat>=0.1.1