lym0302 commited on
Commit
4c6e54c
·
1 Parent(s): 664b2e9
Files changed (1) hide show
  1. reqirments.txt +4 -4
reqirments.txt CHANGED
@@ -1,7 +1,7 @@
1
- torch==2.6.0
2
- torchaudio==2.6.0
3
- torchvision==0.21.0
4
- flash-attn==2.5.8 --no-build-isolation
5
  huggingface_hub >= 0.26
6
  cython
7
  gitpython >= 3.1
 
1
+ torch==2.5.1
2
+ torchaudio==2.5.1
3
+ torchvision==0.20.1
4
+ flash-attn==2.5.8
5
  huggingface_hub >= 0.26
6
  cython
7
  gitpython >= 3.1