shenyunhang commited on
Commit
7e1deea
·
verified ·
1 Parent(s): 2de2c29

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -1
requirements.txt CHANGED
@@ -1,9 +1,10 @@
1
- torch
2
  torchvision
3
  #flash-attn
4
  https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.4.post1/flash_attn-2.7.4.post1+cu12torch2.6cxx11abiTRUE-cp310-cp310-linux_x86_64.whl
5
  deepspeed
6
  accelerate
 
7
  moviepy
8
  decord
9
  pillow
 
1
+ torch==2.6.0
2
  torchvision
3
  #flash-attn
4
  https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.4.post1/flash_attn-2.7.4.post1+cu12torch2.6cxx11abiTRUE-cp310-cp310-linux_x86_64.whl
5
  deepspeed
6
  accelerate
7
+ transformers
8
  moviepy
9
  decord
10
  pillow