prithivMLmods commited on
Commit
4b52060
·
verified ·
1 Parent(s): 789fd6f

update requirements

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -2
requirements.txt CHANGED
@@ -1,11 +1,11 @@
1
- flash-attn @ https://huggingface.co/strangertoolshf/flash_attention_2_wheelhouse/resolve/main/wheelhouse-flash_attn-2.8.3/linux_x86_64/torch2.6/cu12/abiFALSE/cp310/flash_attn-2.8.3+cu12torch2.6cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
2
  git+https://github.com/huggingface/transformers.git
3
  git+https://github.com/huggingface/accelerate.git
4
  git+https://github.com/huggingface/peft.git
5
  huggingface_hub
6
  sentencepiece
7
  opencv-python
8
- torch==2.6.0
9
  torchvision
10
  matplotlib
11
  markdown
 
1
+ flash-attn @ https://huggingface.co/strangertoolshf/flash_attention_2_wheelhouse/resolve/main/wheelhouse-flash_attn-2.8.3/linux_x86_64/torch2.8/cu12/abiFALSE/cp310/flash_attn-2.8.3+cu12torch2.8cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
2
  git+https://github.com/huggingface/transformers.git
3
  git+https://github.com/huggingface/accelerate.git
4
  git+https://github.com/huggingface/peft.git
5
  huggingface_hub
6
  sentencepiece
7
  opencv-python
8
+ torch==2.8.0
9
  torchvision
10
  matplotlib
11
  markdown