AIdeaText commited on
Commit
232b6e5
·
verified ·
1 Parent(s): 1aaec00

Update requirements.txt

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