AIdeaText commited on
Commit
12f4845
1 Parent(s): a9f310e

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +7 -6
requirements.txt CHANGED
@@ -1,16 +1,17 @@
1
  # Extra index for CUDA-enabled PyTorch
2
- --extra-index-url https://download.pytorch.org/whl/cu113
3
 
4
  # Core dependencies
5
- torch>=2.0.0
6
  transformers>=4.36.0
7
- accelerate>=0.21.0
8
  streamlit>=1.24.0
9
- safetensors>=0.3.1
10
  huggingface_hub>=0.19.0
11
 
12
- # Optimization
13
- bitsandbytes>=0.41.0
 
14
  sentencepiece>=0.1.99
15
 
16
  # UI components
 
1
  # Extra index for CUDA-enabled PyTorch
2
+ --extra-index-url https://download.pytorch.org/whl/cu118
3
 
4
  # Core dependencies
5
+ torch>=2.1.0
6
  transformers>=4.36.0
7
+ accelerate>=0.25.0
8
  streamlit>=1.24.0
9
+ safetensors>=0.4.0
10
  huggingface_hub>=0.19.0
11
 
12
+ # Optimization and quantization
13
+ bitsandbytes>=0.41.3
14
+ scipy>=1.9.0
15
  sentencepiece>=0.1.99
16
 
17
  # UI components