yakilee commited on
Commit
4a392fc
·
verified ·
1 Parent(s): 5f78787

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +6 -7
requirements.txt CHANGED
@@ -1,9 +1,8 @@
1
- # Base dependencies that must be installed first
2
- --constraint=https://raw.githubusercontent.com/pytorch/pytorch/master/torch/csrc/empty_file.txt
3
- torch
4
- torchvision
5
 
6
- # Secondary dependencies
7
  numpy==1.24.4
8
  Pillow==10.3.0
9
  Requests==2.31.0
@@ -13,6 +12,6 @@ qwen-vl-utils
13
  gradio==4.44.1
14
  bitsandbytes>=0.41.0
15
 
16
- # Complex build dependencies last
17
- --no-deps
18
  flash-attn==2.5.5
 
1
+ # Core dependencies
2
+ torch>=2.0.0
3
+ torchvision>=0.15.0
 
4
 
5
+ # Other dependencies
6
  numpy==1.24.4
7
  Pillow==10.3.0
8
  Requests==2.31.0
 
12
  gradio==4.44.1
13
  bitsandbytes>=0.41.0
14
 
15
+ # Pre-built wheel for flash-attn
16
+ --find-links https://github.com/Dao-AILab/flash-attention/releases/download/v2.5.5/
17
  flash-attn==2.5.5