Wenye He commited on
Commit
7d2f2eb
·
verified ·
1 Parent(s): e584742

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +7 -6
requirements.txt CHANGED
@@ -1,6 +1,7 @@
1
- torch>=2.1.2
2
- transformers>=4.40.0
3
- gradio>=4.24.0
4
- accelerate>=0.29.0
5
- bitsandbytes>=0.43.0
6
- flash-attn>=2.5.6 # Needed for Phi-3 optimization
 
 
1
+ --extra-index-url https://download.pytorch.org/whl/cu121
2
+ torch==2.1.2+cu121
3
+ flash-attn==2.7.3.post1
4
+ transformers==4.42.0
5
+ gradio==4.29.0
6
+ accelerate==0.30.0
7
+ bitsandbytes==0.43.1