chenjoya commited on
Commit
a60e78b
1 Parent(s): afdabc6

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -3
requirements.txt CHANGED
@@ -1,9 +1,9 @@
1
  --index-url https://download.pytorch.org/whl/cu121
2
- wheel
3
  torch==2.3.1
4
  torchvision==0.18.1
5
  torchaudio==2.3.1
 
 
6
  huggingface_hub==0.23.3
7
  transformers==4.41.2
8
- peft==0.11.1
9
- https://github.com/Dao-AILab/flash-attention/releases/download/v2.5.9.post1/flash_attn-2.5.9.post1+cu122torch2.3cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
 
1
  --index-url https://download.pytorch.org/whl/cu121
 
2
  torch==2.3.1
3
  torchvision==0.18.1
4
  torchaudio==2.3.1
5
+ wheel==0.43.0
6
+ https://github.com/Dao-AILab/flash-attention/releases/download/v2.5.9.post1/flash_attn-2.5.9.post1+cu122torch2.3cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
7
  huggingface_hub==0.23.3
8
  transformers==4.41.2
9
+ peft==0.11.1