Wenye He commited on
Commit
cfb24bd
·
verified ·
1 Parent(s): 0ffa0b9

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +5 -7
requirements.txt CHANGED
@@ -1,7 +1,5 @@
1
- --extra-index-url https://download.pytorch.org/whl/cu121
2
- torch==2.1.2+cu121
3
- flash-attn
4
- transformers==4.42.0
5
- gradio==4.29.0
6
- accelerate==0.30.0
7
- bitsandbytes==0.43.1
 
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