abiabidali commited on
Commit
d71bad6
·
verified ·
1 Parent(s): dc2acc7

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -3
requirements.txt CHANGED
@@ -1,8 +1,7 @@
1
- torch==2.0.1+cu118 # Adjust CUDA version as needed
2
  gradio
3
  transformers
4
  Pillow
5
  timm
6
- # Install flash-attn without attempting to build it
7
- flash-attn --no-build-isolation --global-option="--skip-cuda-build"
8
 
 
1
+ torch==2.0.1+cu118 # Adjust the version based on your system
2
  gradio
3
  transformers
4
  Pillow
5
  timm
6
+
 
7