JeffreyXiang commited on
Commit
79521c8
·
verified ·
1 Parent(s): fbdc8f9

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -2
requirements.txt CHANGED
@@ -2,7 +2,7 @@
2
 
3
  torch==2.10.0
4
  torchvision==0.25.0
5
- triton==3.2.0
6
  pillow==12.0.0
7
  imageio==2.37.2
8
  imageio-ffmpeg==0.6.0
@@ -14,4 +14,5 @@ transformers==4.57.3
14
  zstandard==0.25.0
15
  kornia==0.8.2
16
  timm==1.0.22
17
- git+https://github.com/EasternJournalist/utils3d.git@9a4eb15e4021b67b12c460c7057d642626897ec8
 
 
2
 
3
  torch==2.10.0
4
  torchvision==0.25.0
5
+ triton==3.6.0
6
  pillow==12.0.0
7
  imageio==2.37.2
8
  imageio-ffmpeg==0.6.0
 
14
  zstandard==0.25.0
15
  kornia==0.8.2
16
  timm==1.0.22
17
+ git+https://github.com/EasternJournalist/utils3d.git@9a4eb15e4021b67b12c460c7057d642626897ec8
18
+ https://github.com/mjun0812/flash-attention-prebuild-wheels/releases/download/v0.9.9/flash_attn_3-3.0.0+cu126torch2.10gite2743ab-cp39-abi3-win_amd64.whl