fengyutong commited on
Commit
85a4407
·
verified ·
1 Parent(s): b94267e

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -1
requirements.txt CHANGED
@@ -7,4 +7,5 @@ einops
7
  omegaconf
8
  safetensors
9
  torch==2.5.1
10
- torchvision==0.20.1
 
 
7
  omegaconf
8
  safetensors
9
  torch==2.5.1
10
+ torchvision==0.20.1
11
+ flash-attn==2.6.3