requirements-test.txt 231 Bytes
Newer Older
1
diffusers
zbian's avatar
zbian committed
2
pytest
3
torchvision
4
transformers
5
timm
6
titans
7
torchaudio
8
torchrec
9
contexttimer
oahzxl's avatar
oahzxl committed
10
11
12
einops
triton==2.0.0.dev20221011
git+https://github.com/HazyResearch/flash-attention.git@c422fee3776eb3ea24e011ef641fd5fbeb212623#egg=flash_attn