gokaygokay commited on
Commit
b7084f4
·
verified ·
1 Parent(s): e64598e

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -2
requirements.txt CHANGED
@@ -10,6 +10,6 @@ tokenizers
10
  diffusers
11
  sentencepiece
12
  timm
13
- torch==2.2.0
14
- torchvision==0.17.0
15
  https://github.com/Dao-AILab/flash-attention/releases/download/v2.8.2/flash_attn-2.8.2+cu12torch2.4cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
 
10
  diffusers
11
  sentencepiece
12
  timm
13
+ torch
14
+ torchvision
15
  https://github.com/Dao-AILab/flash-attention/releases/download/v2.8.2/flash_attn-2.8.2+cu12torch2.4cxx11abiFALSE-cp310-cp310-linux_x86_64.whl