abiabidali commited on
Commit
b5d2078
·
verified ·
1 Parent(s): 8f32b3e

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -2
requirements.txt CHANGED
@@ -1,9 +1,10 @@
1
  gradio
2
  transformers
3
- torch
4
  Pillow
5
  timm
6
- flash-attn
 
7
 
8
 
9
 
 
1
  gradio
2
  transformers
3
+ torch==2.5.1 # Ensure this version is compatible with other packages
4
  Pillow
5
  timm
6
+ flash-attn==2.7.0.post2 # Ensure this is compatible with your version of torch
7
+
8
 
9
 
10