KaiChen1998 commited on
Commit
5999d67
·
1 Parent(s): 5f870ca

fix torch version

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -2
requirements.txt CHANGED
@@ -1,11 +1,11 @@
1
  # requirements.txt records the full set of dependencies for development
2
- torch==2.6.0
3
  accelerate
4
  codetiming
5
  datasets
6
  dill
7
  # flash-attn
8
- https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.4.post1/flash_attn-2.7.4.post1+cu12torch2.6cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
9
  hydra-core
10
  liger-kernel
11
  numpy
 
1
  # requirements.txt records the full set of dependencies for development
2
+ torch
3
  accelerate
4
  codetiming
5
  datasets
6
  dill
7
  # flash-attn
8
+ # https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.4.post1/flash_attn-2.7.4.post1+cu12torch2.6cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
9
  hydra-core
10
  liger-kernel
11
  numpy