keynes42 commited on
Commit
ff1cf41
·
verified ·
1 Parent(s): 2bfeb8e

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -1
requirements.txt CHANGED
@@ -1,9 +1,9 @@
1
- flash-attn
2
  gradio
3
  requests
4
  PyYAML
5
  smolagents[toolkit] @ git+https://github.com/huggingface/smolagents.git@v1.16.1
6
  transformers>=4.38.0
 
7
  Pillow
8
  duckduckgo-search
9
  wikipedia-api
 
 
1
  gradio
2
  requests
3
  PyYAML
4
  smolagents[toolkit] @ git+https://github.com/huggingface/smolagents.git@v1.16.1
5
  transformers>=4.38.0
6
+ https://github.com/Dao-AILab/flash-attention/releases/download/v2.5.8/flash_attn-2.5.8+cu122torch2.3cxx11-cp310-cp310-linux_x86_64.whl
7
  Pillow
8
  duckduckgo-search
9
  wikipedia-api